mirror of https://github.com/hpcaitech/ColossalAI
[diffusion] fix package conflicts (#1875)
parent
610dda676c
commit
f9e7d179f2
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue