diff --git a/sana/sana_1600M/config.yaml b/sana/sana_1600M/config.yaml
index 3ab90408..b3cfbf75 100644
--- a/sana/sana_1600M/config.yaml
+++ b/sana/sana_1600M/config.yaml
@@ -13,7 +13,7 @@ model_metadata:
     "num_inference_steps": 18,
     "seed": 4096,
   }
-model_name: sana
+model_name: Sana 1600M
 python_version: py311
 requirements:
 - git+https://github.com/NVlabs/Sana.git@d7945026d8d85008aca1d1e6db5717a1069f5c84
diff --git a/sana/sana_600M/config.yaml b/sana/sana_600M/config.yaml
index 3ab90408..fb9afda1 100644
--- a/sana/sana_600M/config.yaml
+++ b/sana/sana_600M/config.yaml
@@ -13,7 +13,7 @@ model_metadata:
     "num_inference_steps": 18,
     "seed": 4096,
   }
-model_name: sana
+model_name: Sana 600M
 python_version: py311
 requirements:
 - git+https://github.com/NVlabs/Sana.git@d7945026d8d85008aca1d1e6db5717a1069f5c84
diff --git a/sana/sana_600M/model/model.py b/sana/sana_600M/model/model.py
index 44e87236..69ada98f 100644
--- a/sana/sana_600M/model/model.py
+++ b/sana/sana_600M/model/model.py
@@ -50,7 +50,7 @@ def load(self):
         )
         logging.info("loading sana model from hf checkpoint")
         self.sana.from_pretrained(
-            "hf://Efficient-Large-Model/Sana_1600M_1024px/checkpoints/Sana_1600M_1024px.pth"
+            "hf://Efficient-Large-Model/Sana_600M_1024px/checkpoints/Sana_600M_1024px_MultiLing.pth"
         )
         logging.info("all done loading models")
         try: