- 29 Oct, 2022 1 commit
-
-
Pedro Cuenca authored
Tests: upgrade PyTorch cuda to 11.7. Otherwise the cuda versions of torch and torchvision mismatch, and examples tests fail. We were requesting cuda 11.6 for PyTorch, and the default torchvision (via setup.py). Another option would be to include torchvision in the same pip install line as torch.
-
- 26 Oct, 2022 1 commit
-
-
Pi Esposito authored
* add method to enable cuda with minimal gpu usage to stable diffusion * add test to minimal cuda memory usage * ensure all models but unet are onn torch.float32 * move to cpu_offload along with minor internal changes to make it work * make it test against accelerate master branch * coming back, its official: I don't know how to make it test againt the master branch from accelerate * make it install accelerate from master on tests * go back to accelerate>=0.11 * undo prettier formatting on yml files * undo prettier formatting on yml files againn
-
- 03 Oct, 2022 1 commit
-
-
Anton Lozhkov authored
* [CI] Localize the HF cache * pip cache * de-env * refactor matrix * fix fast cache * less onnx steps * revert * revert pip cache * revert pip cache * remove debugging trigger
-
- 21 Sep, 2022 1 commit
-
-
Anton Lozhkov authored
* Add smoke tests for the training examples * upd * use a dummy dataset * mark as slow * cleanup * Update test cases * naming
-
- 19 Sep, 2022 1 commit
-
-
Anton Lozhkov authored
* make_reports * add test utils * style * style
-
- 31 Aug, 2022 2 commits
-
-
Anton Lozhkov authored
-
Anton Lozhkov authored
-
- 30 Aug, 2022 1 commit
-
-
Anton Lozhkov authored
Add datasets + transformers to test deps
-
- 26 Aug, 2022 1 commit
-
-
Anton Lozhkov authored
* Init CI * clarify cpu * style * Check scripts quality too * Drop smi for cpu tests * Run PR tests on cpu docker envs * Update .github/workflows/push_tests.yml Co-authored-by:
Patrick von Platen <patrick.v.platen@gmail.com> * Try minimal python container * Print env, install stable GPU torch * Manual torch install * remove deprecated platform.dist() Co-authored-by:
Patrick von Platen <patrick.v.platen@gmail.com>
-