TechDirty commited on
Commit
d47a98d
·
verified ·
1 Parent(s): 67076f0

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +19 -12
model_index.json CHANGED
@@ -1,13 +1,20 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.18.0",
4
- "components": {
5
- "unet": "unet",
6
- "vae": "vae",
7
- "text_encoder": "text_encoder",
8
- "tokenizer": "tokenizer",
9
- "scheduler": "scheduler",
10
- "safety_checker": "safety_checker"
11
- },
12
- "framework": "pytorch"
13
- }
 
 
 
 
 
 
 
 
1
  {
2
+ "_class_name": "FluxTransformer2DModel",
3
+ "_diffusers_version": "0.33.0.dev0",
4
+ "_name_or_path": "black-forest-labs/FLUX.1-dev",
5
+ "attention_head_dim": 128,
6
+ "axes_dims_rope": [
7
+ 16,
8
+ 56,
9
+ 56
10
+ ],
11
+ "guidance_embeds": true,
12
+ "in_channels": 128,
13
+ "joint_attention_dim": 4096,
14
+ "num_attention_heads": 24,
15
+ "num_layers": 19,
16
+ "num_single_layers": 38,
17
+ "out_channels": 64,
18
+ "patch_size": 1,
19
+ "pooled_projection_dim": 768
20
+ }