Skip to content

Commit dcb0fc6

Browse files
committed
Fix attention mode selection for single file loader
1 parent 24ded5c commit dcb0fc6

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

nodes.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def loadmodel(self, model, base_precision, quantization,
197197
sd = load_torch_file(model_path, device=offload_device, safe_load=True)
198198

199199
with init_empty_weights():
200-
transformer = HunyuanVideoTransformer3DModelPacked(**config)
200+
transformer = HunyuanVideoTransformer3DModelPacked(**config, attention_mode=attention_mode)
201201

202202
params_to_keep = {"norm", "bias", "time_in", "vector_in", "guidance_in", "txt_in", "img_in"}
203203
if quantization == "fp8_e4m3fn" or quantization == "fp8_e4m3fn_fast" or quantization == "fp8_scaled":
@@ -390,7 +390,10 @@ def process(self, model, shift, positive, negative, latent_window_size, use_teac
390390
latent_padding_size = latent_padding * latent_window_size
391391

392392
if embed_interpolation == "linear":
393-
frac = 1 - i / (total_latent_sections - 1) # going backwards
393+
if total_latent_sections <= 1:
394+
frac = 1.0 # Handle case with only one section
395+
else:
396+
frac = 1 - i / (total_latent_sections - 1) # going backwards
394397
else:
395398
frac = start_embed_strength if has_end_image else 1.0
396399

0 commit comments

Comments
 (0)