github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-23 06:00:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

[diffusion] fix package conflicts (#1875)

Browse Source
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/diffusion/README.md
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 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 - you can change the `--logdir` the save the log information and the last checkpoint

4
examples/images/diffusion/environment.yaml
View File

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