fffiloni commited on
Commit
d713a89
·
verified ·
1 Parent(s): 8ca9d5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def main(video_path, audio_path, progress=gr.Progress(track_tqdm=True)):
136
  unet, _ = UNet3DConditionModel.from_pretrained(
137
  OmegaConf.to_container(config.model),
138
  inference_ckpt_path, # load checkpoint
139
- device="cpu",
140
  )
141
 
142
  unet = unet.to(dtype=torch.float16)
 
136
  unet, _ = UNet3DConditionModel.from_pretrained(
137
  OmegaConf.to_container(config.model),
138
  inference_ckpt_path, # load checkpoint
139
+ #device="cpu",
140
  )
141
 
142
  unet = unet.to(dtype=torch.float16)