Skip to content

Commit be518db

Browse files
authored
Remove extraneous clip missing warnings when loading LTX2 embeddings_connector weights (Comfy-Org#11874)
1 parent 80441eb commit be518db

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

comfy/text_encoders/lt.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,9 @@ def load_sd(self, sd):
118118
sdo = comfy.utils.state_dict_prefix_replace(sd, {"text_embedding_projection.aggregate_embed.weight": "text_embedding_projection.weight", "model.diffusion_model.video_embeddings_connector.": "video_embeddings_connector.", "model.diffusion_model.audio_embeddings_connector.": "audio_embeddings_connector."}, filter_keys=True)
119119
if len(sdo) == 0:
120120
sdo = sd
121-
122-
return self.load_state_dict(sdo, strict=False)
121+
missing, unexpected = self.load_state_dict(sdo, strict=False)
122+
missing = [k for k in missing if not k.startswith("gemma3_12b.")] # filter out keys that belong to the main gemma model
123+
return (missing, unexpected)
123124

124125
def memory_estimation_function(self, token_weight_pairs, device=None):
125126
constant = 6.0

0 commit comments

Comments
 (0)