grbell commited on
Commit
52d4b38
·
verified ·
1 Parent(s): ee613ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
35
  torch_dtype=torch.bfloat16
36
  )
37
 
38
- pipe.to("cuda")
39
 
40
  pipe.vae.enable_tiling()
41
  pipe.vae.enable_slicing()
 
35
  torch_dtype=torch.bfloat16
36
  )
37
 
38
+ pipe.to("cpu")
39
 
40
  pipe.vae.enable_tiling()
41
  pipe.vae.enable_slicing()