github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 01:28:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

[example] add diffusion inference (#1986)

Browse Source
This commit is contained in:
Fazzie-Maqianli
2022-11-20 18:35:29 +08:00
committed by GitHub
parent a01278e810
commit b5dbb46172
7 changed files with 343 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
examples/images/diffusion/scripts/download_first_stages.sh Normal file → Executable file
View File

0
examples/images/diffusion/scripts/download_models.sh Normal file → Executable file
View File

6
examples/images/diffusion/scripts/txt2img.sh Executable file
View File

@@ -0,0 +1,6 @@
python scripts/txt2img.py --prompt "Teyvat, Name:Layla, Element: Cryo, Weapon:Sword, Region:Sumeru, Model type:Medium Female, Description:a woman in a blue outfit holding a sword" --plms \
--outdir ./output \
--config /home/lcmql/data2/Genshin/2022-11-18T16-38-46_train_colossalai_teyvattest/checkpoints/last.ckpt \
--ckpt /home/lcmql/data2/Genshin/2022-11-18T16-38-46_train_colossalai_teyvattest/configs/2022-11-18T16-38-46-project.yaml \
--n_samples 4