e7e0cd97ce
* use latest torchao, gradio, schedule-free * get info on slow tests * speed up tests by avoiding gradient checkpointing and reducing eval size
7 lines
271 B
Bash
Executable File
7 lines
271 B
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
# only run one test at a time so as not to OOM the GPU
|
|
pytest -v --durations=10 -n2 /workspace/axolotl/tests/e2e/multigpu/ --ignore=/workspace/axolotl/tests/e2e/multigpu/solo/
|
|
pytest -v --durations=10 -n1 /workspace/axolotl/tests/e2e/multigpu/solo/
|