comi-logic/Wan2GP
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more fixes

Browse Source
This commit is contained in:
DeepBeepMeep
2025-05-17 20:59:46 +02:00
parent ca9e9b94cb
commit 5ba133cae9
3 changed files with 36 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ltx_video/ltxv.py
View File

@@ -154,8 +154,8 @@ class LTXV:
mixed_precision_transformer = False
):
if dtype == torch.float16:
dtype = torch.bfloat16
# 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 ]
@@ -169,8 +169,8 @@ class LTXV:
# vae = CausalVideoAutoencoder.from_pretrained(ckpt_path)
vae = offload.fast_load_transformers_model("ckpts/ltxv_0.9.7_VAE.safetensors", modelClass=CausalVideoAutoencoder)
if VAE_dtype == torch.float16:
VAE_dtype = torch.bfloat16
# if VAE_dtype == torch.float16:
VAE_dtype = torch.bfloat16
vae = vae.to(VAE_dtype)
vae._model_dtype = VAE_dtype