liray commited on
Commit
cf28fff
1 Parent(s): a61fc84
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -198,8 +198,9 @@ sam_predictor = sam_init()
198
  model = model_init()
199
  pipe = StableVideoDiffusionPipeline.from_pretrained(
200
  "stabilityai/stable-video-diffusion-img2vid",
201
- torch_dtype=torch.float16, variant="fp16"
202
- ).to("cuda")
 
203
 
204
 
205
  @spaces.GPU(duration=10)
 
198
  model = model_init()
199
  pipe = StableVideoDiffusionPipeline.from_pretrained(
200
  "stabilityai/stable-video-diffusion-img2vid",
201
+ torch_dtype=torch.float16, variant="fp16", device="cpu"
202
+ )
203
+ pipe = pipe.to("cuda")
204
 
205
 
206
  @spaces.GPU(duration=10)