 dbaranchuk
|
66ce6d9669
make it compatible with low_cpu_mem_usage=True
|
hace 3 años |
|
Dmitry Baranchuk
|
04a2b6f5e3
Support various backend dtypes & async serialization (#38)
|
hace 3 años |
 Artem Chumachenko
|
d989b94614
Pack of Inference Changes (#37)
|
hace 3 años |
|
Dmitry Baranchuk
|
6573076883
Sequential and parallel forward / backward (#36)
|
hace 3 años |
|
Artem Chumachenko
|
6ee942e915
Add GenerationMixin class (#29)
|
hace 3 años |
 justheuristic
|
e2711a033b
Add automated tests (#23)
|
hace 3 años |
|
Dmitry Baranchuk
|
f5463812ad
Shallow prompt tuning (#22)
|
hace 3 años |
|
dbaranchuk
|
21e1f42f04
mv set_requires_grad to remote_model
|
hace 3 años |
|
dbaranchuk
|
79280c4371
refactoring
|
hace 3 años |
|
dbaranchuk
|
6bffeff0a1
fix
|
hace 3 años |
|
dbaranchuk
|
b3cc9e0d99
add LM head for DistributedBloomCausalLM
|
hace 3 años |
|
dbaranchuk
|
df42822f26
LM head for CausalLM & chunked forward
|
hace 3 años |
|
justheuristic
|
1c68670d06
it works
|
hace 3 años |
|
justheuristic
|
88c1bf9896
black-isort
|
hace 3 años |
|
justheuristic
|
4695071ad2
WIP: make DistributedBloom compliant with HF interface
|
hace 3 años |
|
justheuristic
|
e32208c954
black-isort
|
hace 3 años |
|
justheuristic
|
4ad845bce3
black-isort
|
hace 3 años |
|
Dmitry Baranchuk
|
e66ab6f1f2
design interface & refactoring
|
hace 3 años |
 Dmitry Baranchuk
|
d969172208
set requires_grad=False, lm_layer -> h @ word_embeddings, rm lm_layer from comverted_model
|
hace 3 años |
|
justheuristic
|
331591c915
less intrusive warnings
|
hace 3 años |
|
justheuristic
|
9c492bbe8c
Infer prefix by defaukt
|
hace 3 años |
|
justheuristic
|
19ae71e8fc
from_pretrained
|
hace 3 años |
|
justheuristic
|
471e47c0f5
black-isort
|
hace 3 años |
|
justheuristic
|
7d68f6b9a4
fix model creation
|
hace 3 años |
|
justheuristic
|
5849cea28c
prototype remote sequential
|
hace 3 años |
