Upload ./unet/config.json with huggingface_hub
Browse files- unet/config.json +5 -10
unet/config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_class_name": "
|
3 |
"_diffusers_version": "0.27.2",
|
4 |
"act_fn": "silu",
|
5 |
"add_attention": true,
|
@@ -37,15 +37,10 @@
|
|
37 |
"mid_block_scale_factor": 1.41421356237,
|
38 |
"min_decay": 0.0,
|
39 |
"norm_eps": 1e-06,
|
40 |
-
"norm_num_groups":
|
41 |
-
"
|
42 |
-
"num_class_embeds_object_hue": 11,
|
43 |
-
"num_class_embeds_orientation": 16,
|
44 |
-
"num_class_embeds_scale": 9,
|
45 |
-
"num_class_embeds_shape": 5,
|
46 |
-
"num_class_embeds_wall_hue": 11,
|
47 |
"num_train_timesteps": null,
|
48 |
-
"optimization_step":
|
49 |
"out_channels": 3,
|
50 |
"power": 0.75,
|
51 |
"resnet_time_scale_shift": "default",
|
@@ -64,4 +59,4 @@
|
|
64 |
"update_after_step": 0,
|
65 |
"upsample_type": "conv",
|
66 |
"use_ema_warmup": true
|
67 |
-
}
|
|
|
1 |
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
"_diffusers_version": "0.27.2",
|
4 |
"act_fn": "silu",
|
5 |
"add_attention": true,
|
|
|
37 |
"mid_block_scale_factor": 1.41421356237,
|
38 |
"min_decay": 0.0,
|
39 |
"norm_eps": 1e-06,
|
40 |
+
"norm_num_groups": null,
|
41 |
+
"num_class_embeds": 11,
|
|
|
|
|
|
|
|
|
|
|
42 |
"num_train_timesteps": null,
|
43 |
+
"optimization_step": 180000,
|
44 |
"out_channels": 3,
|
45 |
"power": 0.75,
|
46 |
"resnet_time_scale_shift": "default",
|
|
|
59 |
"update_after_step": 0,
|
60 |
"upsample_type": "conv",
|
61 |
"use_ema_warmup": true
|
62 |
+
}
|