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

fixed LTX Video support for RTX10xx , 20xx

Browse Source
This commit is contained in:
DeepBeepMeep
2025-05-17 23:28:48 +02:00
parent 5ba133cae9
commit 8b9929f744
3 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
hyvideo/text_encoder/__init__.py
View File

@@ -189,7 +189,7 @@ class TextEncoder(nn.Module):
if "llm" in text_encoder_type:
from mmgp import offload
forcedConfigPath= None if "i2v" in text_encoder_type else "ckpts/llava-llama-3-8b/config.json"
self.model= offload.fast_load_transformers_model(self.model_path, forcedConfigPath=forcedConfigPath)
self.model= offload.fast_load_transformers_model(self.model_path, modelPrefix="language_model" if forcedConfigPath != None else None, forcedConfigPath=forcedConfigPath)
if forcedConfigPath != None:
self.model.final_layer_norm = self.model.model.norm