Hugging Face's logo

Spaces:
Omnibus
/
Video-Diffusion-WebUI
Runtime error

App Files Community
Omnibus commited on
Commit
03af647
·
1 Parent(s): 16aba50

Update video_diffusion/inpaint_zoom/zoom_in_app.py

Browse files
Files changed (1) hide show
  1. video_diffusion/inpaint_zoom/zoom_in_app.py +1 -2
video_diffusion/inpaint_zoom/zoom_in_app.py CHANGED
@@ -13,8 +13,7 @@ if torch.cuda.is_available():
13
  dtype = torch.float16
14
  else:
15
  device = torch.device("cpu")
16
- dtype= torch.HalfTensor
17
-
18
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
19
 
20
 
 
13
  dtype = torch.float16
14
  else:
15
  device = torch.device("cpu")
16
+ dtype= torch.get_default_dtype()
 
17
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
18
 
19