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
Alexander Borzunov 8c546d988a
Test Llama, rebalancing, throughput eval, and all CLI scripts (#452)
This PR extends CI to:

1. Test Llama code using [TinyLlama-v0](https://huggingface.co/Maykeye/TinyLLama-v0).
2. Test rebalancing (sets up a situation where the 1st server needs to change its original position).
3. Check if benchmark scripts run (in case someone breaks its code). Note that the benchmark results are meaningless here (since they're measured on a tiny swarm of CPU servers, with low `--n_steps`).
4. Test `petals.cli.run_dht`.
5. Increase swap space and watch free RAM (a common issue is that actions are cancelled without explanation if there's not enough RAM - so it's a useful reminder + debug tool).
6. Fix flapping tests for bloom-560m by increasing tolerance.

Other minor changes: fix `--help` messages to show defaults, fix docs, tune rebalancing constants.
10 months ago
..
bootstrap.id Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
conftest.py Fix logging: do not duplicate lines, enable colors in Colab (#156) 1 year ago
server2.id Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_aux_functions.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_block_exact_match.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_chained_calls.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_dtype.py Add LLaMA support (#323) 11 months ago
test_full_model.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_peft.py Support peft LoRA adapters (#335) 11 months ago
test_priority_pool.py Fix issues related to `petals` as a module (#159) 1 year ago
test_remote_sequential.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_sequence_manager.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_server_stats.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_tensor_parallel.py Test Llama, rebalancing, throughput eval, and all CLI scripts (#452) 10 months ago
test_utils.py Support peft LoRA adapters (#335) 11 months ago