Upload ./unet/config.json with huggingface_hub
Browse files- unet/config.json +8 -1
unet/config.json
CHANGED
|
@@ -16,6 +16,7 @@
|
|
| 16 |
],
|
| 17 |
"center_input_sample": true,
|
| 18 |
"class_embed_type": null,
|
|
|
|
| 19 |
"down_block_types": [
|
| 20 |
"SkipDownBlock2D",
|
| 21 |
"SkipDownBlock2D",
|
|
@@ -31,8 +32,10 @@
|
|
| 31 |
"flip_sin_to_cos": true,
|
| 32 |
"freq_shift": 0,
|
| 33 |
"in_channels": 3,
|
|
|
|
| 34 |
"layers_per_block": 2,
|
| 35 |
"mid_block_scale_factor": 1.41421356237,
|
|
|
|
| 36 |
"norm_eps": 1e-06,
|
| 37 |
"norm_num_groups": 32,
|
| 38 |
"num_class_embeds_floor_hue": 11,
|
|
@@ -42,7 +45,9 @@
|
|
| 42 |
"num_class_embeds_shape": 5,
|
| 43 |
"num_class_embeds_wall_hue": 11,
|
| 44 |
"num_train_timesteps": null,
|
|
|
|
| 45 |
"out_channels": 3,
|
|
|
|
| 46 |
"resnet_time_scale_shift": "default",
|
| 47 |
"sample_size": 64,
|
| 48 |
"set_W_to_weight": false,
|
|
@@ -56,5 +61,7 @@
|
|
| 56 |
"SkipUpBlock2D",
|
| 57 |
"SkipUpBlock2D"
|
| 58 |
],
|
| 59 |
-
"
|
|
|
|
|
|
|
| 60 |
}
|
|
|
|
| 16 |
],
|
| 17 |
"center_input_sample": true,
|
| 18 |
"class_embed_type": null,
|
| 19 |
+
"decay": 0.9999,
|
| 20 |
"down_block_types": [
|
| 21 |
"SkipDownBlock2D",
|
| 22 |
"SkipDownBlock2D",
|
|
|
|
| 32 |
"flip_sin_to_cos": true,
|
| 33 |
"freq_shift": 0,
|
| 34 |
"in_channels": 3,
|
| 35 |
+
"inv_gamma": 1.0,
|
| 36 |
"layers_per_block": 2,
|
| 37 |
"mid_block_scale_factor": 1.41421356237,
|
| 38 |
+
"min_decay": 0.0,
|
| 39 |
"norm_eps": 1e-06,
|
| 40 |
"norm_num_groups": 32,
|
| 41 |
"num_class_embeds_floor_hue": 11,
|
|
|
|
| 45 |
"num_class_embeds_shape": 5,
|
| 46 |
"num_class_embeds_wall_hue": 11,
|
| 47 |
"num_train_timesteps": null,
|
| 48 |
+
"optimization_step": 70000,
|
| 49 |
"out_channels": 3,
|
| 50 |
+
"power": 0.75,
|
| 51 |
"resnet_time_scale_shift": "default",
|
| 52 |
"sample_size": 64,
|
| 53 |
"set_W_to_weight": false,
|
|
|
|
| 61 |
"SkipUpBlock2D",
|
| 62 |
"SkipUpBlock2D"
|
| 63 |
],
|
| 64 |
+
"update_after_step": 0,
|
| 65 |
+
"upsample_type": "conv",
|
| 66 |
+
"use_ema_warmup": true
|
| 67 |
}
|