Hugging Face's logo

nvidia
/
Cosmos3-Nano-Policy-DROID

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

Add modular pipeline index

#4
by atharvajoshi10 - opened
base: refs/heads/main
←
from: refs/pr/4
Discussion Files changed
+61
-0
Files changed (1) hide show
  1. modular_model_index.json +61 -0
modular_model_index.json ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_blocks_class_name": "Cosmos3OmniBlocks",
3
+ "_class_name": "Cosmos3OmniModularPipeline",
4
+ "_diffusers_version": "0.39.0.dev0",
5
+ "text_tokenizer": [
6
+ "transformers",
7
+ "Qwen2TokenizerFast",
8
+ {
9
+ "pretrained_model_name_or_path": "nvidia/Cosmos3-Nano-Policy-DROID",
10
+ "revision": null,
11
+ "subfolder": "text_tokenizer",
12
+ "type_hint": [
13
+ "transformers",
14
+ "Qwen2TokenizerFast"
15
+ ],
16
+ "variant": null
17
+ }
18
+ ],
19
+ "vae": [
20
+ "diffusers",
21
+ "AutoencoderKLWan",
22
+ {
23
+ "pretrained_model_name_or_path": "nvidia/Cosmos3-Nano-Policy-DROID",
24
+ "revision": null,
25
+ "subfolder": "vae",
26
+ "type_hint": [
27
+ "diffusers",
28
+ "AutoencoderKLWan"
29
+ ],
30
+ "variant": null
31
+ }
32
+ ],
33
+ "transformer": [
34
+ "diffusers",
35
+ "Cosmos3OmniTransformer",
36
+ {
37
+ "pretrained_model_name_or_path": "nvidia/Cosmos3-Nano-Policy-DROID",
38
+ "revision": null,
39
+ "subfolder": "transformer",
40
+ "type_hint": [
41
+ "diffusers",
42
+ "Cosmos3OmniTransformer"
43
+ ],
44
+ "variant": null
45
+ }
46
+ ],
47
+ "scheduler": [
48
+ "diffusers",
49
+ "UniPCMultistepScheduler",
50
+ {
51
+ "pretrained_model_name_or_path": "nvidia/Cosmos3-Nano-Policy-DROID",
52
+ "revision": null,
53
+ "subfolder": "scheduler",
54
+ "type_hint": [
55
+ "diffusers",
56
+ "UniPCMultistepScheduler"
57
+ ],
58
+ "variant": null
59
+ }
60
+ ]
61
+ }