Model load error with newest comfyUI

[SaiZyca]

ELLALoader Error(s) in loading state_dict for ELLAModel: Invalid key(s) in state_dict: "connector.latents", "connector.perceiver_blocks.0.attn.in_proj_bias", "connector.perceiver_blocks.0.attn.in_proj_weight", "connector.perceiver_blocks.0.attn.out_proj.bias", "connector.perceiver_blocks.0.attn.out_proj.weight", "connector.perceiver_blocks.0.ln_1.linear.bias", "connector.perceiver_blocks.0.ln_1.linear.weight", "connector.perceiver_blocks.0.ln_2.linear.bias", "connector.perceiver_blocks.0.ln_2.linear.weight", "connector.perceiver_blocks.0.ln_ff.linear.bias", "connector.perceiver_blocks.0.ln_ff.linear.weight", "connector.perceiver_blocks.0.mlp.c_fc.bias", "connector.perceiver_blocks.0.mlp.c_fc.weight", "connector.perceiver_blocks.0.mlp.c_proj.bias", "connector.perceiver_blocks.0.mlp.c_proj.weight", "connector.perceiver_blocks.1.attn.in_proj_bias", "connector.perceiver_blocks.1.attn.in_proj_weight", "connector.perceiver_blocks.1.attn.out_proj.bias", "connector.perceiver_blocks.1.attn.out_proj.weight", "connector.perceiver_blocks.1.ln_1.linear.bias", "connector.perceiver_blocks.1.ln_1.linear.weight", "connector.perceiver_blocks.1.ln_2.linear.bias", "connector.perceiver_blocks.1.ln_2.linear.weight", "connector.perceiver_blocks.1.ln_ff.linear.bias", "connector.perceiver_blocks.1.ln_ff.linear.weight", "connector.perceiver_blocks.1.mlp.c_fc.bias", "connector.perceiver_blocks.1.mlp.c_fc.weight", "connector.perceiver_blocks.1.mlp.c_proj.bias", "connector.perceiver_blocks.1.mlp.c_proj.weight", "connector.perceiver_blocks.2.attn.in_proj_bias", "connector.perceiver_blocks.2.attn.in_proj_weight", "connector.perceiver_blocks.2.attn.out_proj.bias", "connector.perceiver_blocks.2.attn.out_proj.weight", "connector.perceiver_blocks.2.ln_1.linear.bias", "connector.perceiver_blocks.2.ln_1.linear.weight", "connector.perceiver_blocks.2.ln_2.linear.bias", "connector.perceiver_blocks.2.ln_2.linear.weight", "connector.perceiver_blocks.2.ln_ff.linear.bias", "connector.perceiver_blocks.2.ln_ff.linear.weight", "connector.perceiver_blocks.2.mlp.c_fc.bias", "connector.perceiver_blocks.2.mlp.c_fc.weight", "connector.perceiver_blocks.2.mlp.c_proj.bias", "connector.perceiver_blocks.2.mlp.c_proj.weight", "connector.perceiver_blocks.3.attn.in_proj_bias", "connector.perceiver_blocks.3.attn.in_proj_weight", "connector.perceiver_blocks.3.attn.out_proj.bias", "connector.perceiver_blocks.3.attn.out_proj.weight", "connector.perceiver_blocks.3.ln_1.linear.bias", "connector.perceiver_blocks.3.ln_1.linear.weight", "connector.perceiver_blocks.3.ln_2.linear.bias", "connector.perceiver_blocks.3.ln_2.linear.weight", "connector.perceiver_blocks.3.ln_ff.linear.bias", "connector.perceiver_blocks.3.ln_ff.linear.weight", "connector.perceiver_blocks.3.mlp.c_fc.bias", "connector.perceiver_blocks.3.mlp.c_fc.weight", "connector.perceiver_blocks.3.mlp.c_proj.bias", "connector.perceiver_blocks.3.mlp.c_proj.weight", "connector.perceiver_blocks.4.attn.in_proj_bias", "connector.perceiver_blocks.4.attn.in_proj_weight", "connector.perceiver_blocks.4.attn.out_proj.bias", "connector.perceiver_blocks.4.attn.out_proj.weight", "connector.perceiver_blocks.4.ln_1.linear.bias", "connector.perceiver_blocks.4.ln_1.linear.weight", "connector.perceiver_blocks.4.ln_2.linear.bias", "connector.perceiver_blocks.4.ln_2.linear.weight", "connector.perceiver_blocks.4.ln_ff.linear.bias", "connector.perceiver_blocks.4.ln_ff.linear.weight", "connector.perceiver_blocks.4.mlp.c_fc.bias", "connector.perceiver_blocks.4.mlp.c_fc.weight", "connector.perceiver_blocks.4.mlp.c_proj.bias", "connector.perceiver_blocks.4.mlp.c_proj.weight", "connector.perceiver_blocks.5.attn.in_proj_bias", "connector.perceiver_blocks.5.attn.in_proj_weight", "connector.perceiver_blocks.5.attn.out_proj.bias", "connector.perceiver_blocks.5.attn.out_proj.weight", "connector.perceiver_blocks.5.ln_1.linear.bias", "connector.perceiver_blocks.5.ln_1.linear.weight", "connector.perceiver_blocks.5.ln_2.linear.bias", "connector.perceiver_blocks.5.ln_2.linear.weight", "connector.perceiver_blocks.5.ln_ff.linear.bias", "connector.perceiver_blocks.5.ln_ff.linear.weight", "connector.perceiver_blocks.5.mlp.c_fc.bias", "connector.perceiver_blocks.5.mlp.c_fc.weight", "connector.perceiver_blocks.5.mlp.c_proj.bias", "connector.perceiver_blocks.5.mlp.c_proj.weight", "connector.proj_in.bias", "connector.proj_in.weight", "connector.time_aware_linear.bias", "connector.time_aware_linear.weight", "time_embedding.linear_1.bias", "time_embedding.linear_1.weight", "time_embedding.linear_2.bias", "time_embedding.linear_2.weight", mismatched dtypes or shape.

[dchatel]

Change ComfyUI-ELLA/model.py line 319 from load_model(self.model, path, strict=True) to load_model(self.model, path, strict=False)

[DavideAlidosi]

Change ComfyUI-ELLA/model.py line 319 from load_model(self.model, path, strict=True) to load_model(self.model, path, strict=False)

Thanks a lot! Bests.