bowong
/
Wan2GP
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed LTX Video FP16

This commit is contained in:
DeepBeepMeep 2025-05-17 18:43:31 +02:00
parent 2a2853a893
commit 2dd9260197
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ltx_video/ltxv.py
View File

@ -154,6 +154,8 @@ class LTXV:
mixed_precision_transformer = False mixed_precision_transformer = False
): ):
if dtype == torch.float16:
dtype = torch.bfloat16
self.mixed_precision_transformer = mixed_precision_transformer self.mixed_precision_transformer = mixed_precision_transformer
self.distilled = any("lora" in name for name in model_filepath) self.distilled = any("lora" in name for name in model_filepath)
model_filepath = [name for name in model_filepath if not "lora" in name ] model_filepath = [name for name in model_filepath if not "lora" in name ]