diff --git a/tank/pytorch/v_diffusion_pytorch/setup_v_diffusion_pytorch.sh b/tank/pytorch/v_diffusion_pytorch/setup_v_diffusion_pytorch.sh
index d6ec340533..a1d6ef1b00 100644
--- 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/