From 2dd926019710ac14fa1fdf0958d513eb508e196d Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Sat, 17 May 2025 18:43:31 +0200
Subject: [PATCH] fixed LTX Video FP16

---
 ltx_video/ltxv.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ltx_video/ltxv.py b/ltx_video/ltxv.py
index 7be215b..e0decbb 100644
--- a/ltx_video/ltxv.py
+++ b/ltx_video/ltxv.py
@@ -154,6 +154,8 @@ class LTXV:
         mixed_precision_transformer = False
     ):
 
+        if dtype == torch.float16:
+            dtype  = torch.bfloat16
         self.mixed_precision_transformer = mixed_precision_transformer
         self.distilled = any("lora" in name for name in model_filepath)
         model_filepath = [name for name in model_filepath if not "lora" in name ]