theunlikely commited on
Commit
f94dd31
·
verified ·
1 Parent(s): e81fb8f

Upload 2 files

Browse files
Files changed (2) hide show
  1. comfy_config.json +25 -0
  2. config.json +14 -0
comfy_config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_class": "Flux",
3
+ "model_config": {
4
+ "axes_dim": [
5
+ 16,
6
+ 56,
7
+ 56
8
+ ],
9
+ "context_in_dim": 4096,
10
+ "depth": 19,
11
+ "depth_single_blocks": 38,
12
+ "disable_unet_model_creation": true,
13
+ "guidance_embed": true,
14
+ "hidden_size": 3072,
15
+ "image_model": "flux",
16
+ "in_channels": 16,
17
+ "mlp_ratio": 4.0,
18
+ "num_heads": 24,
19
+ "out_channels": 16,
20
+ "patch_size": 2,
21
+ "qkv_bias": true,
22
+ "theta": 10000,
23
+ "vec_in_dim": 768
24
+ }
25
+ }
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FluxTransformer2DModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "../checkpoints/flux-dev/transformer",
5
+ "attention_head_dim": 128,
6
+ "guidance_embeds": true,
7
+ "in_channels": 64,
8
+ "joint_attention_dim": 4096,
9
+ "num_attention_heads": 24,
10
+ "num_layers": 19,
11
+ "num_single_layers": 38,
12
+ "patch_size": 1,
13
+ "pooled_projection_dim": 768
14
+ }