Unverified Commit f9e7d179 authored by HELSON's avatar HELSON Committed by GitHub
Browse files

[diffusion] fix package conflicts (#1875)

parent 610dda67
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment