giulio98 commited on
Commit
4691720
·
verified ·
1 Parent(s): 72ffa0e

Upload ./unet/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. unet/config.json +1 -8
unet/config.json CHANGED
@@ -16,7 +16,6 @@
16
  ],
17
  "center_input_sample": true,
18
  "class_embed_type": null,
19
- "decay": 0.9999,
20
  "down_block_types": [
21
  "SkipDownBlock2D",
22
  "SkipDownBlock2D",
@@ -32,10 +31,8 @@
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,9 +42,7 @@
45
  "num_class_embeds_shape": 5,
46
  "num_class_embeds_wall_hue": 11,
47
  "num_train_timesteps": null,
48
- "optimization_step": 180000,
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,7 +56,5 @@
61
  "SkipUpBlock2D",
62
  "SkipUpBlock2D"
63
  ],
64
- "update_after_step": 0,
65
- "upsample_type": "conv",
66
- "use_ema_warmup": true
67
  }
 
16
  ],
17
  "center_input_sample": true,
18
  "class_embed_type": null,
 
19
  "down_block_types": [
20
  "SkipDownBlock2D",
21
  "SkipDownBlock2D",
 
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
  "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
  "SkipUpBlock2D",
57
  "SkipUpBlock2D"
58
  ],
59
+ "upsample_type": "conv"
 
 
60
  }