fix repo name bug

2025-03-04 20:06:32 +08:00
parent b507c699ef
commit 2ffe8fa9bd
1 changed files with 1 additions and 1 deletions
@@ -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