From 7dccb42beffe6d6e35dce461aa889055ddb9abab Mon Sep 17 00:00:00 2001 From: Enze Xie Date: Thu, 21 Nov 2024 18:03:33 +0800 Subject: [PATCH] Update README.md. Signed-off-by: junsongc --- README.md | 3 ++- sana/tools/hf_utils.py | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index a465004..2c47224 100644 --- a/README.md +++ b/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 diff --git a/sana/tools/hf_utils.py b/sana/tools/hf_utils.py index 04ec993..3b3ae8c 100644 --- a/sana/tools/hf_utils.py +++ b/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, ): """