[diffusion] fix package conflicts (#1875)

pull/1859/head^2
HELSON 2022-11-10 16:33:34 +08:00 committed by GitHub
parent 610dda676c
commit f9e7d179f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -63,7 +63,7 @@ we provide the script `train.sh` to run the training task , and two Stategy in `
for example, you can run the training from colossalai by
```
python main.py --logdir /tmp -t --postfix test -b config/train_colossalai.yaml
python main.py --logdir /tmp -t --postfix test -b configs/train_colossalai.yaml
```
- you can change the `--logdir` the save the log information and the last checkpoint

View File

@ -17,14 +17,14 @@ dependencies:
- invisible-watermark
- imageio==2.9.0
- imageio-ffmpeg==0.4.2
- pytorch-lightning==1.4.2
- pytorch-lightning==1.8.0
- omegaconf==2.1.1
- test-tube>=0.7.5
- streamlit>=0.73.1
- einops==0.3.0
- torch-fidelity==0.3.0
- transformers==4.19.2
- torchmetrics==0.6.0
- torchmetrics==0.7.0
- kornia==0.6
- prefetch_generator
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers