Shrink tiny Cosmos3 transformer for CPU tests

transformer/config.json

@@ -1,7 +1,6 @@
 {
   "_class_name": "Cosmos3OmniTransformer",
   "_diffusers_version": "0.40.0.dev0",
-  "_name_or_path": "/tmp/cosmos3-tiny-modular-prototype-20260715",
   "action_dim": 10,
   "action_gen": true,
   "attention_bias": false,
@@ -9,27 +8,27 @@
   "backbone_type": "cosmos3_qwen3vl",
   "base_fps": 24,
   "enable_fps_modulation": true,
-  "head_dim": 8,
+  "head_dim": 6,
   "hidden_act": "silu",
-  "hidden_size": 32,
-  "intermediate_size": 64,
+  "hidden_size": 6,
+  "intermediate_size": 12,
   "latent_channel": 4,
   "latent_patch_size": 2,
-  "num_attention_heads": 4,
+  "num_attention_heads": 1,
   "num_embodiment_domains": 32,
-  "num_hidden_layers": 2,
-  "num_key_value_heads": 2,
+  "num_hidden_layers": 1,
+  "num_key_value_heads": 1,
   "patch_latent_dim": 16,
   "qk_norm_for_text": true,
   "rms_norm_eps": 1e-06,
   "rope_axes_dim": [
-    2,
+    1,
     1,
     1
   ],
   "rope_scaling": {
     "mrope_section": [
-      2,
+      1,
       1,
       1
     ]

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:db737dd3c20b7b31d0e71972de97177cb4839b16c199b213c8f380ba144b5fe5
-size 39109792
+oid sha256:382866740d429a75e2ca32857f115ab8d5ce3bc430ff0e4d0191c9afb6a7f1ce
+size 7311624