Hugging Face's logo

Spaces:
fffiloni
/
DimensionX
Running

App Files Community
8
fffiloni commited on
Commit
937e122
·
verified ·
1 Parent(s): 6b5b440

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ hf_hub_download(
22
  local_dir="checkpoints"
23
  )
24
 
25
- pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-5b", torch_dtype=torch.bfloat16).to(device)
26
  pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
27
 
28
  pipe_image = CogVideoXImageToVideoPipeline.from_pretrained(
 
22
  local_dir="checkpoints"
23
  )
24
 
25
+ pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-5b", torch_dtype=torch.bfloat16)
26
  pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
27
 
28
  pipe_image = CogVideoXImageToVideoPipeline.from_pretrained(