From 2ffe8fa9bd77eeaba063d26f2d06b22789d71d0f Mon Sep 17 00:00:00 2001
From: chflame163 <chflame@163.com>
Date: Tue, 4 Mar 2025 20:06:32 +0800
Subject: [PATCH] fix repo name bug

---
 py/cogview4.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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