diff --git a/py/cogview4.py b/py/cogview4.py
index 61a6063..14005cc 100644
--- a/py/cogview4.py
+++ b/py/cogview4.py
@@ -59,7 +59,7 @@ class CogView4_Wrapper:
         from diffusers import CogView4Pipeline
 
         if self.dtype != dtype or self.model_name != model:
-            model_path = check_and_download_model(model, f"THUDD/{model}")
+            model_path = check_and_download_model(model, f"THUDM/{model}")
             if dtype == 'bf16':
                 self.pipe = CogView4Pipeline.from_pretrained(model_path, torch_dtype=torch.bfloat16).to("cuda")
                 self.model_name = model