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
Pavel Samygin 0be21775af
remove transformer block, implement as sequential of size 1 (#54)
* remove transformer block, implement as sequence size 1
* reimplement get_remote_module
* fix readme

Co-authored-by: Alexander Borzunov <hxrussia@gmail.com>
Co-authored-by: Aleksandr Borzunov <borzunov.alexander@gmail.com>
2 years ago
..
bloom [Fix] make distributed seq cls to not create the full bloom model (#49) 2 years ago
client remove transformer block, implement as sequential of size 1 (#54) 2 years ago
server Let users specify sequence length instead of assuming 2048 (#52) 2 years ago
utils Convert actual model weights (#46) 2 years ago
__init__.py Measure and cache network & compute throughput (#21) 2 years ago
data_structures.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2 years ago
dht_utils.py remove transformer block, implement as sequential of size 1 (#54) 2 years ago