Update open_oasis_master/generate.py

open_oasis_master/generate.py

@@ -97,7 +97,7 @@ for i in tqdm(range(n_prompt_frames, total_frames)):
 
         # get model predictions
         with torch.no_grad():
-            with autocast("cuda", dtype=torch.half):
+            with autocast("cpu", dtype=torch.half):
                 v = model(x_curr, t, actions[:, start_frame : i + 1])
 
         x_start = alphas_cumprod[t].sqrt() * x_curr - (1 - alphas_cumprod[t]).sqrt() * v