Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|