liray commited on
Commit
a61fc84
·
1 Parent(s): 708612a
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -197,11 +197,9 @@ def model_init():
197
  sam_predictor = sam_init()
198
  model = model_init()
199
  pipe = StableVideoDiffusionPipeline.from_pretrained(
200
- "/scratch/shared/beegfs/ruining/projects/generative-models/stable-video-diffusion-img2vid",
201
  torch_dtype=torch.float16, variant="fp16"
202
- )
203
- pipe.vae.to(dtype=torch.float16, device="cuda")
204
- pipe.image_encoder = pipe.image_encoder.to("cuda")
205
 
206
 
207
  @spaces.GPU(duration=10)
 
197
  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)