From f9e7d179f29990002ccb27f626890a81ff67e743 Mon Sep 17 00:00:00 2001 From: HELSON Date: Thu, 10 Nov 2022 16:33:34 +0800 Subject: [PATCH] [diffusion] fix package conflicts (#1875) --- examples/images/diffusion/README.md | 2 +- examples/images/diffusion/environment.yaml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/images/diffusion/README.md b/examples/images/diffusion/README.md index 38878ab71..06459bfe5 100644 --- a/examples/images/diffusion/README.md +++ b/examples/images/diffusion/README.md @@ -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 diff --git a/examples/images/diffusion/environment.yaml b/examples/images/diffusion/environment.yaml index fc529102c..79b706b83 100644 --- a/examples/images/diffusion/environment.yaml +++ b/examples/images/diffusion/environment.yaml @@ -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