zRzRzRzRzRzRzR commited on
Commit
dc70128
1 Parent(s): 5564375

remove wrong path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("/share/home/zyx/Models/cogvideox-5b-0823-hf", torch_dtype=dtype).to(device)
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)