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/tests
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
..
scripts Reduce vocabulary size in test model, fix bug in routing when overlapped (#45) 2 years ago
conftest.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2 years ago
test.id Add automated tests (#23) 2 years ago
test_block_exact_match.py remove transformer block, implement as sequential of size 1 (#54) 2 years ago
test_chained_calls.py remove transformer block, implement as sequential of size 1 (#54) 2 years ago
test_full_model.py Let users specify sequence length instead of assuming 2048 (#52) 2 years ago
test_remote_sequential.py Deep distributed prompt tuning (#42) 2 years ago
test_utils.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2 years ago