* [Studio] Install flash attn at setup time for linux * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * cleanup changes Signed-off-by: Datta Nimmaturi <venkatadattasainimmaturi@gmail.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Test cases * wheel_utils: narrow url_exists exceptions and log at debug level --------- Signed-off-by: Datta Nimmaturi <venkatadattasainimmaturi@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Roland Tannous <115670425+rolandtannous@users.noreply.github.com> Co-authored-by: Roland Tannous <rolandtannous@gravityq.ai> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| conftest.py | ||
| test_anthropic_messages.py | ||
| test_cache_case_resolution.py | ||
| test_data_recipe_seed.py | ||
| test_gpu_selection.py | ||
| test_gpu_selection_sandbox.py | ||
| test_kv_cache_estimation.py | ||
| test_models_get_model_config_case_resolution.py | ||
| test_native_context_length.py | ||
| test_responses_api.py | ||
| test_studio_api.py | ||
| test_trained_model_scan.py | ||
| test_training_worker_flash_attn.py | ||
| test_transformers_version.py | ||
| test_utils.py | ||
| test_vision_cache.py | ||
| test_vram_estimation.py | ||