Update setup_v_diffusion_pytorch.sh (nod-ai#291)

Fix minor issue with v-diffusion PyTorch version
danielschulz · Aug 24, 2022 · f79a6bf · f79a6bf
1 parent 7dc27a7
commit f79a6bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tank/pytorch/v_diffusion_pytorch/setup_v_diffusion_pytorch.sh b/tank/pytorch/v_diffusion_pytorch/setup_v_diffusion_pytorch.sh
@@ -20,3 +20,5 @@ pip install -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html -
 
 mkdir checkpoints
 wget https://the-eye.eu/public/AI/models/v-diffusion/cc12m_1_cfg.pth -P checkpoints/
+
+cp -r checkpoints/ v-diffusion-pytorch/