github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[tank][v-diffusion] Polish up v-diffusion UX (#315)

Browse Source
This commit is contained in:
Stanley Winata
2022-09-08 12:55:51 -07:00
committed by GitHub
parent 18872222d3
commit 95ec3d7216
3 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tank/pytorch/v_diffusion_pytorch/.gitignore vendored Normal file
View File

@@ -0,0 +1,4 @@
*.png
*.pth
checkpoints/
v-diffusion-pytorch/

0
tank/pytorch/v_diffusion_pytorch/cfg_sample_preprocess.py Normal file → Executable file
View File

3
tank/pytorch/v_diffusion_pytorch/setup_v_diffusion_pytorch.sh Normal file → Executable file
View File

@@ -1,3 +1,5 @@
#!/bin/bash
TD="$(cd $(dirname $0) && pwd)"
if [ -z "$PYTHON" ]; then
PYTHON="$(which python3)"
@@ -22,3 +24,4 @@ mkdir checkpoints
wget https://the-eye.eu/public/AI/models/v-diffusion/cc12m_1_cfg.pth -P checkpoints/
cp -r checkpoints/ v-diffusion-pytorch/
cp cc12m_1.py v-diffusion-pytorch/diffusion/models/.