Hugging Face's logo

nvidia
/
Cosmos3-Edge-Policy-DROID

Cosmos
Diffusers
Safetensors
cosmos3_omni
nvidia
cosmos3
world action model
policy model
Model card Files
xet
 Community
5

Add Cosmos3 Edge scheduler metadata

#10
by atharvajoshi10 - opened
base: refs/heads/main
←
from: refs/pr/10
Discussion Files changed
+11
-6
Files changed (3) hide show
  1. model_index.json +3 -2
  2. modular_model_index.json +7 -3
  3. scheduler/scheduler_config.json +1 -1
model_index.json CHANGED
@@ -9,7 +9,7 @@
9
  ],
10
  "scheduler": [
11
  "diffusers",
12
- "UniPCMultistepScheduler"
13
  ],
14
  "sound_tokenizer": [
15
  null,
@@ -27,5 +27,6 @@
27
  "vae": [
28
  "diffusers",
29
  "AutoencoderKLWan"
30
- ]
 
31
  }
 
9
  ],
10
  "scheduler": [
11
  "diffusers",
12
+ "Cosmos3EdgeUniPCMultistepScheduler"
13
  ],
14
  "sound_tokenizer": [
15
  null,
 
27
  "vae": [
28
  "diffusers",
29
  "AutoencoderKLWan"
30
+ ],
31
+ "native_flow_shift": 5.0
32
  }
modular_model_index.json CHANGED
@@ -5,13 +5,13 @@
5
  "is_distilled": false,
6
  "scheduler": [
7
  "diffusers",
8
- "UniPCMultistepScheduler",
9
  {
10
  "pretrained_model_name_or_path": "nvidia/Cosmos3-Edge-Policy-DROID",
11
  "subfolder": "scheduler",
12
  "type_hint": [
13
  "diffusers",
14
- "UniPCMultistepScheduler"
15
  ],
16
  "variant": null
17
  }
@@ -54,5 +54,9 @@
54
  ],
55
  "variant": null
56
  }
57
- ]
 
 
 
 
58
  }
 
5
  "is_distilled": false,
6
  "scheduler": [
7
  "diffusers",
8
+ "Cosmos3EdgeUniPCMultistepScheduler",
9
  {
10
  "pretrained_model_name_or_path": "nvidia/Cosmos3-Edge-Policy-DROID",
11
  "subfolder": "scheduler",
12
  "type_hint": [
13
  "diffusers",
14
+ "Cosmos3EdgeUniPCMultistepScheduler"
15
  ],
16
  "variant": null
17
  }
 
54
  ],
55
  "variant": null
56
  }
57
+ ],
58
+ "default_use_system_prompt": false,
59
+ "enable_safety_checker": false,
60
+ "use_native_flow_schedule": true,
61
+ "native_flow_shift": 5.0
62
  }
scheduler/scheduler_config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "UniPCMultistepScheduler",
3
  "_diffusers_version": "0.40.0.dev0",
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",
 
1
  {
2
+ "_class_name": "Cosmos3EdgeUniPCMultistepScheduler",
3
  "_diffusers_version": "0.40.0.dev0",
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",