diff --git a/README.md b/README.md
index 709a5b3..f6c280a 100644
--- a/README.md
+++ b/README.md
@@ -17,7 +17,7 @@ $ conda activate zecon
 Our source code relies on [blended diffusion](https://github.com/omriav/blended-diffusion).
 
 ### Pre-trained model
-Download the model weights trained on [imagenet](https://github.com/openai/guided-diffusion) and [ffhq](https://github.com/jychoi118/ilvr_adm) dataset, respectively.
+Download the model weights trained on [imagenet](https://github.com/openai/guided-diffusion) (256x256_diffusion_uncond.pt) and [ffhq](https://github.com/jychoi118/ilvr_adm) (ffhq_baseline.pt) dataset, respectively.
 
 Create a folder ```'./ckpt/'``` and then place the downloaded weights into the folder.