You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
petals/src/petals/models
Artem Chumachenko d2fcbbc72e
Add Mixtral models (#553)
* Add somehow workable version

* Fix generation

* Fixes

* Choose right attn

* style

* fix bloom

* remove unnes

* Update src/petals/models/mixtral/model.py

Co-authored-by: Max Ryabinin <mryabinin0@gmail.com>

* fix order of init

---------

Co-authored-by: Max Ryabinin <mryabinin0@gmail.com>
2 months ago
..
bloom Bump transformers and accelerate versions (#554) 4 months ago
falcon Add position_ids argument to DistributedFalconModel (#525) 8 months ago
llama Fix p2p pushing in rpc_inference (by @miaoqijun ) , support transformers 4.38.2 (#563) 3 months ago
mixtral Add Mixtral models (#553) 2 months ago
__init__.py Add Mixtral models (#553) 2 months ago