From edf2ca8b5db21289184845f93bfe823fcbc7a801 Mon Sep 17 00:00:00 2001
From: Silky Singh <54901747+silky1708@users.noreply.github.com>
Date: Thu, 4 Jan 2024 16:43:01 +0530
Subject: [PATCH] add specific checkpoints to download

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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.