 Aleksandr Borzunov
|
a4d0c9e82f
Fix
|
hace 2 años |
|
Aleksandr Borzunov
|
51d96edfa7
Fix
|
hace 2 años |
|
Aleksandr Borzunov
|
dcf5183b69
Don't use .lm_head() in benchmark_forward.py
|
hace 2 años |
|
Aleksandr Borzunov
|
cf87264199
Hardcode more initial peers
|
hace 2 años |
|
Aleksandr Borzunov
|
88251e100c
MAX_TOKENS_IN_BATCH = 512
|
hace 2 años |
|
Aleksandr Borzunov
|
52c1149751
Use dtype float32
|
hace 2 años |
|
Aleksandr Borzunov
|
d5e27c262b
Fix initial peer
|
hace 2 años |
|
Aleksandr Borzunov
|
4721094b84
Add benchmark_forward.py
|
hace 2 años |
|
Aleksandr Borzunov
|
205eb2f2d8
Add initial peer, show speed
|
hace 2 años |
|
Aleksandr Borzunov
|
84776bff73
Mark required args
|
hace 2 años |
|
Aleksandr Borzunov
|
96ad0e0cc3
chmod +x benchmark_inference.py
|
hace 2 años |
|
Aleksandr Borzunov
|
5092b35171
Add inference benchmark
|
hace 2 años |
|
Aleksandr Borzunov
|
2cfd70d751
Debug mode: load empty block
|
hace 2 años |
 Alexander Borzunov
|
675bacb592
Bump version to 1.1.5 (#312)
|
hace 2 años |
|
Alexander Borzunov
|
e026952338
Abort speedtest if it runs too long (#316)
|
hace 2 años |
|
Alexander Borzunov
|
6eb306a605
Raise error for unexpected .generate() kwargs (#315)
|
hace 2 años |
|
Alexander Borzunov
|
d9e7bfc949
Divide compute throughput by average no. of used blocks (#314)
|
hace 2 años |
|
Alexander Borzunov
|
6137b1b4b0
Replace .make_sequence(..., mode="random") with mode="max_throughput" (#313)
|
hace 2 años |
|
Alexander Borzunov
|
0a313bf6c5
Update hivemind to 1.1.8, enable efficient bfloat16 encoding (#311)
|
hace 2 años |
|
Alexander Borzunov
|
8f6342a861
Refactor RemoteSequenceManager (#309)
|
hace 2 años |
|
Alexander Borzunov
|
454c193863
Fix OOMs happening in case of accelerate >= 0.16.0 (#310)
|
hace 2 años |
|
Alexander Borzunov
|
93c4eba5d1
Bump version to 1.1.4 (#306)
|
hace 2 años |
|
Alexander Borzunov
|
c0e0e1319d
Force transformers to use config.torch_dtype by default (#307)
|
hace 2 años |
|
Alexander Borzunov
|
98be9ffe4c
Relax the rest of Hugging Face dependencies (#305)
|
hace 2 años |
|
Alexander Borzunov
|
5c0b4286b2
Suggest commands for Docker first (#304)
|
hace 2 años |
|
Alexander Borzunov
|
35662b4a16
Require bitsandbytes == 0.38.0.post2, hivemind == 1.1.7 (#302)
|
hace 2 años |
|
Alexander Borzunov
|
21c3526ec1
Start SequenceManager's thread only after first .make_sequence() (#301)
|
hace 2 años |
|
Alexander Borzunov
|
6c6150f684
Remove use_auto_relay=True in client (#300)
|
hace 2 años |
|
Alexander Borzunov
|
892fa2386a
Remove CustomLinear8bitLt (#297)
|
hace 2 años |
|
Alexander Borzunov
|
74d8cda8c4
Add Python 3.10 to CI (#299)
|
hace 2 años |
