shaaravpawar commited on
Commit
9a3a0d3
·
verified ·
1 Parent(s): 885bde8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from diffusers import DiffusionPipeline
2
  import torch
3
 
4
- device = "cuda" if torch.cuda.is _available() "cpu" else
5
 
6
  pipeline=DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
7
  pipeline = pipeline.to(device)
 
1
  from diffusers import DiffusionPipeline
2
  import torch
3
 
4
+ device = "cuda" if torch.cuda.is _available() else "cpu"
5
 
6
  pipeline=DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
7
  pipeline = pipeline.to(device)