ai-tube-model-ltxv-1

Paused

daniel shalem commited on 23 days ago

Commit

d675fab

•

1 Parent(s): 34518f8

Bugfix: pass device type for autocast.

Files changed (1) hide show

xora/pipelines/pipeline_xora_video.py CHANGED Viewed

@@ -1015,7 +1015,7 @@ class XoraVideoPipeline(DiffusionPipeline):
                             "Mixed precision is not supported yet on XLA devices."
                         )
-                    context_manager = torch.autocast(device, dtype=torch.bfloat16)
                 else:
                     context_manager = nullcontext()  # Dummy context manager

                             "Mixed precision is not supported yet on XLA devices."
                         )
+                    context_manager = torch.autocast(device.type, dtype=torch.bfloat16)
                 else:
                     context_manager = nullcontext()  # Dummy context manager