Update README.md.

Signed-off-by: junsongc <[email protected]>

README.md

@@ -35,6 +35,7 @@ As a result, Sana-0.6B is very competitive with modern giant diffusion model (e.
 
 ## 🔥🔥 News
 
+- (🔥 New) \[2024/11\] 1.6B [Sana models](https://huggingface.co/collections/Efficient-Large-Model/sana-673efba2a57ed99843f11f9e) are released.
 - (🔥 New) \[2024/11\] Training & Inference & Metrics code are released.
 - (🔥 New) \[2024/11\] Working on [`diffusers`](https://github.com/huggingface/diffusers/pull/9982).
 - \[2024/10\] [Demo](https://nv-sana.mit.edu/) is released.
@@ -206,7 +207,7 @@ We will try our best to release
 
 - \[x\] Training code
 - \[x\] Inference code
-- \[ \] Model zoo
+- \[+\] Model zoo
 - \[ \] working on Diffusers(https://github.com/huggingface/diffusers/pull/9982)
 - \[ \] ComfyUI
 - \[ \] Laptop development

sana/tools/hf_utils.py

@@ -33,10 +33,10 @@ def hf_download_or_fpath(path):
 
 
 def hf_download_data(
-    repo_id="Efficient-Large-Model/VILA-inference-demos",
-    filename="imagenet_cat.jpg",
+    repo_id="Efficient-Large-Model/Sana_1600M_1024px",
+    filename="checkpoints/Sana_1600M_1024px.pth",
     cache_dir=None,
-    repo_type="dataset",
+    repo_type="model",
     download_full_repo=False,
 ):
     """