Hugging Face's logo

nvidia
/
Cosmos3-Edge

Cosmos
Diffusers
Safetensors
cosmos3_edge
nvidia
cosmos3
Model card Files
xet
 Community
6

Add Cosmos3 Edge scheduler metadata

#47
by atharvajoshi10 - opened
base: refs/heads/main
←
from: refs/pr/47
Discussion Files changed
+13
-8
Files changed (3) hide show
  1. model_index.json +3 -2
  2. modular_model_index.json +7 -3
  3. scheduler/scheduler_config.json +3 -3
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": 3.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",
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",
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": 3.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",
@@ -7,7 +7,7 @@
7
  "disable_corrector": [],
8
  "dynamic_thresholding_ratio": 0.995,
9
  "final_sigmas_type": "zero",
10
- "flow_shift": 1.0,
11
  "lower_order_final": true,
12
  "num_train_timesteps": 1000,
13
  "predict_x0": true,
@@ -29,5 +29,5 @@
29
  "use_dynamic_shifting": false,
30
  "use_exponential_sigmas": false,
31
  "use_flow_sigmas": true,
32
- "use_karras_sigmas": true
33
  }
 
1
  {
2
+ "_class_name": "Cosmos3EdgeUniPCMultistepScheduler",
3
  "_diffusers_version": "0.40.0.dev0",
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",
 
7
  "disable_corrector": [],
8
  "dynamic_thresholding_ratio": 0.995,
9
  "final_sigmas_type": "zero",
10
+ "flow_shift": 3.0,
11
  "lower_order_final": true,
12
  "num_train_timesteps": 1000,
13
  "predict_x0": true,
 
29
  "use_dynamic_shifting": false,
30
  "use_exponential_sigmas": false,
31
  "use_flow_sigmas": true,
32
+ "use_karras_sigmas": false
33
  }