Wing Lian 2c37bf6c21 Prune cuda117 (#327) ...

* drop cuda117/torch 1.13.1 from support, pin flash attention to v2.0.1, rm torchvision/torchaudio install

* gptq base build not needed. add sm 9.0 support

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

..

Dockerfile

add runpod envs to .bashrc, fix bnb env (#316)

2023-07-22 10:09:38 -04:00

Dockerfile-base

Prune cuda117 (#327)

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

Dockerfile-runpod

set transformers cache env var in docker image

2023-07-16 23:03:54 -04:00