Spaces:

Lightricks
/

LTX-Video-Playground

Running on A100

Nisan Chiprut commited on 3 days ago

Commit

7f8a00d

•

2 Parent(s): 34518f8 d675fab

Merge pull request #40 from LightricksResearch/bugfix/fix-device-type-pass

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