From b158fbdaddf883d7ad960f789154465ceabaf85d Mon Sep 17 00:00:00 2001
From: learningpro <learningpro@163.com>
Date: Tue, 6 Aug 2024 15:16:34 +0800
Subject: [PATCH] Update gradio_demo.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

使用huggingface下载模型路径
---
 gradio_demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gradio_demo.py b/gradio_demo.py
index 65b7b1c..65eeb48 100644
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -17,7 +17,7 @@ import PIL
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = CogVideoXPipeline.from_pretrained("/share/home/zyx/Models/cogvideox-hf-0805", torch_dtype=dtype).to(device)
+pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-2b", torch_dtype=dtype).to(device)
 
 sys_prompt = """You are part of a team of bots that creates videos. You work with an assistant bot that will draw anything you say in square brackets.