Spaces:
Runtime error
Runtime error
zR
commited on
Commit
·
dc70128
1
Parent(s):
5564375
remove wrong path
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ import moviepy.editor as mp
|
|
12 |
|
13 |
dtype = torch.bfloat16
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
-
pipe = CogVideoXPipeline.from_pretrained("/
|
16 |
|
17 |
os.makedirs("./output", exist_ok=True)
|
18 |
os.makedirs("./gradio_tmp", exist_ok=True)
|
|
|
12 |
|
13 |
dtype = torch.bfloat16
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
+
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-2b", torch_dtype=dtype).to(device)
|
16 |
|
17 |
os.makedirs("./output", exist_ok=True)
|
18 |
os.makedirs("./gradio_tmp", exist_ok=True)
|