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
Max Ryabinin bd91be27ea
Add missing methods for SamplingAlgorithm, fix docstrings (#107)
* Add missing methods for SamplingAlgorithm, fix docstrings

* Add SamplingAlgorithm to _choose_sample_algorithm

* Add test_sampling

* Add a warning if sampling options were passed, but do_sample=False

* Skip the sampling test for now

Co-authored-by: Alexander Borzunov <borzunov.alexander@gmail.com>
2 years ago
..
scripts Fix arguments in remove_old_models.py (#153) 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_aux_functions.py Bump transformers to 4.25.1 (#151) 2 years ago
test_block_exact_match.py Make Petals a pip-installable package (attempt 2) (#102) 2 years ago
test_chained_calls.py Make Petals a pip-installable package (attempt 2) (#102) 2 years ago
test_full_model.py Add missing methods for SamplingAlgorithm, fix docstrings (#107) 2 years ago
test_linear8bitlt.py Support --load_in_8bit on pre-Turing GPUs (#113) 2 years ago
test_priority_pool.py Make Petals a pip-installable package (attempt 2) (#102) 2 years ago
test_remote_sequential.py Optimize RemoteSequenceManager (#106) 2 years ago
test_sequence_manager.py Hotfix span selection (#110) 2 years ago
test_utils.py Implement RemoteSequential slicing and extra repr, add tests (#30) 2 years ago