jbilcke-hf
/

LTX-Video-0.9.1-HFIE

Model card Files Files and versions

jbilcke-hf HF Staff commited on Feb 24

Commit

a1ba913

·

verified ·

1 Parent(s): eba7e0a

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -151,7 +151,7 @@ class EndpointHandler:
                 torch_dtype=torch.bfloat16
             ).to("cuda")
-            apply_teacache(self.image_to_video)
         else:
             # Initialize models with bfloat16 precision
@@ -160,7 +160,7 @@ class EndpointHandler:
                 torch_dtype=torch.bfloat16
             ).to("cuda")
-            apply_teacache(self.text_to_video)
         # Initialize LoRA tracking
         self._current_lora_model = None

                 torch_dtype=torch.bfloat16
             ).to("cuda")
+            #apply_teacache(self.image_to_video)
         else:
             # Initialize models with bfloat16 precision
                 torch_dtype=torch.bfloat16
             ).to("cuda")
+            #apply_teacache(self.text_to_video)
         # Initialize LoRA tracking
         self._current_lora_model = None