nodes/kijai-ComfyUI-HunyuanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Probably irrelevant, cleanup

Browse Source
This commit is contained in:
kijai
2025-03-08 12:11:04 +02:00
parent 2559003f40
commit 4bc21fdece
1 changed files with 15 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hyvideo/text_encoder/__init__.py
+15 -21
View File
@@ -410,28 +410,22 @@ class TextEncoder(nn.Module):
else:
last_hidden_state = outputs[self.output_key]
if prompt_template is not None:
if data_type == 'I2V_image':
crop_start = prompt_template.get("crop_start", -1)
crop_end = prompt_template.get('assistant_emb_start', -1)
elif data_type == 'I2V_video':
crop_start = prompt_template.get("crop_start", -1)
text_crop_start = crop_start - 1 + prompt_template.get("image_emb_len", 576)
image_crop_start = prompt_template.get("image_emb_start", 5)
image_crop_end = prompt_template.get('image_emb_end', 581)
batch_indices, last_double_return_token_indices = torch.where(
batch_encoding["input_ids"] == prompt_template.get('double_return_token_id', 271))
last_double_return_token_indices = last_double_return_token_indices.reshape(
batch_encoding["input_ids"].shape[0], -1)[:, -1]
batch_indices = batch_indices.reshape(batch_encoding["input_ids"].shape[0], -1)[:, -1]
assistant_crop_start = last_double_return_token_indices - 1 + prompt_template.get(
"image_emb_len", 576) - 4
assistant_crop_end = last_double_return_token_indices - 1 + prompt_template.get(
"image_emb_len", 576)
crop_start = prompt_template.get("crop_start", -1)
text_crop_start = crop_start - 1 + prompt_template.get("image_emb_len", 576)
image_crop_start = prompt_template.get("image_emb_start", 5)
image_crop_end = prompt_template.get('image_emb_end', 581)
batch_indices, last_double_return_token_indices = torch.where(
batch_encoding["input_ids"] == prompt_template.get('double_return_token_id', 271))
last_double_return_token_indices = last_double_return_token_indices.reshape(
batch_encoding["input_ids"].shape[0], -1)[:, -1]
batch_indices = batch_indices.reshape(batch_encoding["input_ids"].shape[0], -1)[:, -1]
assistant_crop_start = last_double_return_token_indices - 1 + prompt_template.get(
"image_emb_len", 576) - 4
assistant_crop_end = last_double_return_token_indices - 1 + prompt_template.get(
"image_emb_len", 576)
attention_mask_assistant_crop_start = last_double_return_token_indices - 4
attention_mask_assistant_crop_end = last_double_return_token_indices
else:
raise ValueError(f"Unsupported data type: {data_type}")
attention_mask_assistant_crop_start = last_double_return_token_indices - 4
attention_mask_assistant_crop_end = last_double_return_token_indices
text_last_hidden_state = []
text_attention_mask = []