YiYiXu HF Staff commited on
Commit
c049b62
·
verified ·
1 Parent(s): b116921

Upload MellonPipelineConfig

Browse files
Files changed (1) hide show
  1. mellon_pipeline_config.json +158 -0
mellon_pipeline_config.json ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "label": "Flux2 Klein Auto Blocks",
3
+ "default_repo": "",
4
+ "default_dtype": "",
5
+ "node_params": {
6
+ "custom": {
7
+ "params": {
8
+ "prompt": {
9
+ "label": "Prompt",
10
+ "type": "string",
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",
116
+ "type": "image",
117
+ "display": "output"
118
+ },
119
+ "doc": {
120
+ "label": "Doc",
121
+ "type": "string",
122
+ "display": "output"
123
+ }
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",
152
+ "doc"
153
+ ],
154
+ "block_name": "custom",
155
+ "node_type": "custom"
156
+ }
157
+ }
158
+ }