Wing Lian 24146733db E2e device cuda (#575) ...

* use torch.cuda.current_device() instead of local_rank

* ignore NVML errors for gpu stats

* llama lora packing e2e tests

2023-09-14 22:49:27 -04:00

..

base.yml

Prune cuda117 (#327)

2023-07-26 16:27:49 -04:00

e2e.yml

E2e device cuda (#575)

2023-09-14 22:49:27 -04:00

main.yml

e2e testing (#574)

2023-09-14 21:56:11 -04:00

pre-commit.yml

Add pre-commit: black+flake8+pylint

2023-05-31 02:53:22 +09:00

pypi.yml

fix the sed command to replace the version w the tag

2023-09-11 13:44:19 -04:00

tests.yml

e2e testing (#574)

2023-09-14 21:56:11 -04:00