add dpm-solver (mindspore-lab#28)

* add dpm-solver add dpm-solver * add dpm-solver add dpm-solver * add dpm-solver add dpm-solver
wxrui2013 · Dec 28, 2022 · 9e09445 · 9e09445
1 parent 4a647e6
commit 9e09445
Show file tree

Hide file tree

Showing 5 changed files with 1,268 additions and 12 deletions.
diff --git a/vision/wukong-huahua/infer.sh b/vision/wukong-huahua/infer.sh
@@ -18,14 +18,7 @@
 export GLOG_v=3
 export ASCEND_GLOBAL_LOG_LEVEL=3
 export ASCEND_SLOG_PRINT_TO_STDOUT=0
+export DEVICE_ID=0
 
-python txt2img.py \
---prompt "来自深渊 风景 绘画 写实风格" \
---config configs/v1-inference-chinese.yaml \
---output_path ./output/ \
---seed 42 \
---n_iter 4 \
---n_samples 8 \
---W 512 \
---H 512 \
---ddim_steps 50 
+python txt2img.py --prompt "来自深渊 风景 绘画 写实风格" --config configs/v1-inference-chinese.yaml --output_path ./output/ --seed 42 --dpm_solver --n_iter 4 --n_samples 4 --W 512 --H 512 --ddim_steps 15
+#python txt2img.py --prompt "来自深渊 风景 绘画 写实风格" --config configs/v1-inference-chinese.yaml --output_path ./output/ --seed 42 --n_iter 4 --n_samples 4 --W 512 --H 512 --ddim_steps 30
diff --git a/vision/wukong-huahua/ldm/models/diffusion/dpm_solver/__init__.py b/vision/wukong-huahua/ldm/models/diffusion/dpm_solver/__init__.py
@@ -0,0 +1 @@
+from .sampler import DPMSolverSampler