Hugging Face's logo

diffusers
/
FLUX.2-klein-4B-modular

Diffusers
Model card Files
xet
Community
YiYiXu HF Staff commited on
Commit
b577843
·
verified ·
1 Parent(s): f18ec12

Upload MellonPipelineConfig

Browse files
Files changed (1) hide show
  1. mellon_pipeline_config.json +112 -4
mellon_pipeline_config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "label": "Flux2 Klein Auto Blocks",
3
- "default_repo": "YiYiXu/FLUX.2-kelin-4B-modular",
4
- "default_dtype": "bfloat16",
5
  "node_params": {
6
  "custom": {
7
  "params": {
@@ -11,15 +11,105 @@
11
  "display": "textarea",
12
  "default": ""
13
  },
 
 
 
 
 
 
 
 
 
 
14
  "image": {
15
  "label": "Image",
16
  "type": "image",
17
  "display": "input"
18
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  "num_inference_steps": {
20
  "label": "Num Inference Steps",
21
  "type": "int",
22
- "default": 4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  },
24
  "images": {
25
  "label": "Images",
@@ -34,10 +124,28 @@
34
  },
35
  "input_names": [
36
  "prompt",
 
 
37
  "image",
38
- "num_inference_steps"
 
 
 
 
 
 
 
 
 
 
39
  ],
40
  "model_input_names": [
 
 
 
 
 
 
41
  ],
42
  "output_names": [
43
  "images",
 
1
  {
2
  "label": "Flux2 Klein Auto Blocks",
3
+ "default_repo": "",
4
+ "default_dtype": "",
5
  "node_params": {
6
  "custom": {
7
  "params": {
 
11
  "display": "textarea",
12
  "default": ""
13
  },
14
+ "max_sequence_length": {
15
+ "label": "Max Sequence Length",
16
+ "type": "custom",
17
+ "display": "input"
18
+ },
19
+ "text_encoder_out_layers": {
20
+ "label": "Text Encoder Out Layers",
21
+ "type": "custom",
22
+ "display": "input"
23
+ },
24
  "image": {
25
  "label": "Image",
26
  "type": "image",
27
  "display": "input"
28
  },
29
+ "height": {
30
+ "label": "Height",
31
+ "type": "custom",
32
+ "display": "input"
33
+ },
34
+ "width": {
35
+ "label": "Width",
36
+ "type": "custom",
37
+ "display": "input"
38
+ },
39
+ "generator": {
40
+ "label": "Generator",
41
+ "type": "custom",
42
+ "display": "input"
43
+ },
44
+ "num_images_per_prompt": {
45
+ "label": "Num Images Per Prompt",
46
+ "type": "custom",
47
+ "display": "input"
48
+ },
49
+ "image_latents": {
50
+ "label": "Image Latents",
51
+ "type": "custom",
52
+ "display": "input"
53
+ },
54
+ "latents": {
55
+ "label": "Latents",
56
+ "type": "custom",
57
+ "display": "input"
58
+ },
59
  "num_inference_steps": {
60
  "label": "Num Inference Steps",
61
  "type": "int",
62
+ "default": 50
63
+ },
64
+ "timesteps": {
65
+ "label": "Timesteps",
66
+ "type": "custom",
67
+ "display": "input"
68
+ },
69
+ "sigmas": {
70
+ "label": "Sigmas",
71
+ "type": "custom",
72
+ "display": "input"
73
+ },
74
+ "joint_attention_kwargs": {
75
+ "label": "Joint Attention Kwargs",
76
+ "type": "custom",
77
+ "display": "input"
78
+ },
79
+ "output_type": {
80
+ "label": "Output Type",
81
+ "type": "custom",
82
+ "display": "input"
83
+ },
84
+ "text_encoder": {
85
+ "label": "Text Encoder",
86
+ "type": "diffusers_auto_model",
87
+ "display": "input"
88
+ },
89
+ "tokenizer": {
90
+ "label": "Tokenizer",
91
+ "type": "diffusers_auto_model",
92
+ "display": "input"
93
+ },
94
+ "image_processor": {
95
+ "label": "Image Processor",
96
+ "type": "diffusers_auto_model",
97
+ "display": "input"
98
+ },
99
+ "vae": {
100
+ "label": "Vae",
101
+ "type": "diffusers_auto_model",
102
+ "display": "input"
103
+ },
104
+ "scheduler": {
105
+ "label": "Scheduler",
106
+ "type": "diffusers_auto_model",
107
+ "display": "input"
108
+ },
109
+ "transformer": {
110
+ "label": "Transformer",
111
+ "type": "diffusers_auto_model",
112
+ "display": "input"
113
  },
114
  "images": {
115
  "label": "Images",
 
124
  },
125
  "input_names": [
126
  "prompt",
127
+ "max_sequence_length",
128
+ "text_encoder_out_layers",
129
  "image",
130
+ "height",
131
+ "width",
132
+ "generator",
133
+ "num_images_per_prompt",
134
+ "image_latents",
135
+ "latents",
136
+ "num_inference_steps",
137
+ "timesteps",
138
+ "sigmas",
139
+ "joint_attention_kwargs",
140
+ "output_type"
141
  ],
142
  "model_input_names": [
143
+ "text_encoder",
144
+ "tokenizer",
145
+ "image_processor",
146
+ "vae",
147
+ "scheduler",
148
+ "transformer"
149
  ],
150
  "output_names": [
151
  "images",