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>
2022-09-01 04:26:31 +03:00
..
scripts Reduce vocabulary size in test model, fix bug in routing when overlapped (#45) 2022-08-17 18:50:52 +03:00
conftest.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2022-07-19 04:28:04 +03:00
test_block_exact_match.py remove transformer block, implement as sequential of size 1 (#54) 2022-09-01 04:26:31 +03:00
test_chained_calls.py remove transformer block, implement as sequential of size 1 (#54) 2022-09-01 04:26:31 +03:00
test_full_model.py Let users specify sequence length instead of assuming 2048 (#52) 2022-08-29 21:04:37 +03:00
test_remote_sequential.py Deep distributed prompt tuning (#42) 2022-08-12 18:28:21 +03:00
test_utils.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2022-07-19 04:28:04 +03:00
test.id Add automated tests (#23) 2022-07-16 01:59:23 +03:00