Added added_kv_proj_dim for i2v

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline.py

@@ -107,6 +107,8 @@ def create_model(rngs: nnx.Rngs, wan_config: dict):
     wan_config = restored_checkpoint["wan_config"]
   else:
     wan_config = WanModel.load_config(config.pretrained_model_name_or_path, subfolder=subfolder)
+  if config.model_type == "I2V":
+    wan_config["added_kv_proj_dim"] = 1024
   wan_config["mesh"] = mesh
   wan_config["dtype"] = config.activations_dtype
   wan_config["weights_dtype"] = config.weights_dtype