Merge pull request #2120 from Fazziekey/example/stablediffusion-v2

[example] support stable diffusion v2

Merge pull request #2120 from Fazziekey/example/stablediffusion-v2
[example] support stable diffusion v2
6c4c6a04 · Fazzie-Maqianli · GitHub · 5efda697 · cea4292a · 6c4c6a04
Unverified Commit 6c4c6a04 authored Dec 13, 2022 by Fazzie-Maqianli Committed by GitHub Dec 13, 2022
11 changed files
--- a/examples/images/diffusion/ldm/modules/midas/midas/midas_net_custom.py
+++ b/examples/images/diffusion/ldm/modules/midas/midas/midas_net_custom.py
--- a/examples/images/diffusion/ldm/modules/midas/midas/transforms.py
+++ b/examples/images/diffusion/ldm/modules/midas/midas/transforms.py
--- a/examples/images/diffusion/ldm/modules/midas/midas/vit.py
+++ b/examples/images/diffusion/ldm/modules/midas/midas/vit.py
--- a/examples/images/diffusion/ldm/modules/midas/utils.py
+++ b/examples/images/diffusion/ldm/modules/midas/utils.py
--- a/examples/images/diffusion/ldm/modules/x_transformer.py
+++ b/examples/images/diffusion/ldm/modules/x_transformer.py
--- a/examples/images/diffusion/ldm/util.py
+++ b/examples/images/diffusion/ldm/util.py
--- a/examples/images/diffusion/main.py
+++ b/examples/images/diffusion/main.py
--- a/examples/images/diffusion/requirements.txt
+++ b/examples/images/diffusion/requirements.txt
--- a/examples/images/diffusion/scripts/img2img.py
+++ b/examples/images/diffusion/scripts/img2img.py
--- a/examples/images/diffusion/scripts/txt2img.py
+++ b/examples/images/diffusion/scripts/txt2img.py
--- a/examples/images/diffusion/train.sh
+++ b/examples/images/diffusion/train.sh
-HF_DATASETS_OFFLINE=1 
-TRANSFORMERS_OFFLINE=1 
+# HF_DATASETS_OFFLINE=1
+# TRANSFORMERS_OFFLINE=1
+# DIFFUSERS_OFFLINE=1

-python main.py --logdir /tmp -t --postfix test -b configs/train_colossalai.yaml 
+python main.py --logdir /tmp/ -t -b configs/Teyvat/train_colossalai_teyvat.yaml