unittest workflow fix maybe

This commit is contained in:
Victor Hall 2023-04-22 18:56:51 -04:00
parent f6a87eb327
commit b7c7625051
2 changed files with 20 additions and 1 deletions

View File

@ -23,7 +23,7 @@ jobs:
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install -r requirements-test.txt
- name: Run unit tests
run: |

19
requirements-test.txt Normal file
View File

@ -0,0 +1,19 @@
diffusers[torch]>=0.14.0
ninja
numpy
omegaconf==2.2.3
protobuf==3.20.3
pyre-extensions==0.0.23
pytorch-lightning==1.9.2
transformers==4.27.1
aiohttp==3.8.4
bitsandbytes==0.37.2
colorama==0.4.6
compel~=1.1.3
ftfy==6.1.1
lion-pytorch
piexif==1.1.3
pyfakefs
pynvml==11.5.0
tensorboard==2.12.0
wandb