Update checkpoint

model.safetensors.index.json

@@ -143,7 +143,7 @@
     "layers.22.mlp.down_proj.weight": "transformer/diffusion_pytorch_model-00002-of-00002.safetensors",
     "layers.22.mlp.up_proj.weight": "transformer/diffusion_pytorch_model-00002-of-00002.safetensors",
     "layers.22.post_attention_layernorm.weight": "transformer/diffusion_pytorch_model-00002-of-00002.safetensors",
-    "layers.22.self_attn.k_norm_und_for_gen.weight": "transformer/diffusion_pytorch_model-00002-of-00002.safetensors",
+    "layers.22.self_attn.k_norm_und_for_gen.weight": "transformer/diffusion_pytorch_model-00001-of-00002.safetensors",
     "layers.22.self_attn.to_k.weight": "transformer/diffusion_pytorch_model-00001-of-00002.safetensors",
     "layers.22.self_attn.to_out.weight": "transformer/diffusion_pytorch_model-00001-of-00002.safetensors",
     "layers.22.self_attn.to_q.weight": "transformer/diffusion_pytorch_model-00001-of-00002.safetensors",

modular_model_index.json

@@ -3,6 +3,8 @@
   "_class_name": "Cosmos3OmniModularPipeline",
   "_diffusers_version": "0.40.0.dev0",
   "is_distilled": false,
+  "default_use_system_prompt": false,
+  "use_native_flow_schedule": true,
   "scheduler": [
     "diffusers",
     "UniPCMultistepScheduler",

transformer/diffusion_pytorch_model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f74b228d29f844a58bef266f3afc2d695fdc7e00f0d18b618f5889966586891b
-size 5000039696
+oid sha256:116134e5492c514df79a2f07214e577c397aa961387dbe1c5dda488e7994ff6b
+size 5000040072

transformer/diffusion_pytorch_model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:96e9031a940699710dd11b3b81345c22418471ad82bb838abe40f4f03eb86503
-size 1739335880
+oid sha256:3b4c0aafe2705c900ac7bc743085a8e9e09c906624331bc2baa21e172e5d48e4
+size 1739335512

transformer/diffusion_pytorch_model.safetensors.index.json

@@ -305,7 +305,7 @@
     "layers.22.self_attn.add_k_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
     "layers.22.self_attn.add_q_proj.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
     "layers.22.self_attn.add_v_proj.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
-    "layers.22.self_attn.k_norm_und_for_gen.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
+    "layers.22.self_attn.k_norm_und_for_gen.weight": "diffusion_pytorch_model-00001-of-00002.safetensors",
     "layers.22.self_attn.norm_added_k.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
     "layers.22.self_attn.norm_added_q.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",
     "layers.22.self_attn.to_add_out.weight": "diffusion_pytorch_model-00002-of-00002.safetensors",