tolgacangoz commited on
Commit
9fe0cde
·
verified ·
1 Parent(s): 068db56

Update unet/nesting_level_0/config.json

Browse files
Files changed (1) hide show
  1. unet/nesting_level_0/config.json +2 -2
unet/nesting_level_0/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "NestedUNet2DConditionModel",
3
  "_diffusers_version": "0.31.0.dev0",
4
  "_name_or_path": "/home/cosmos/Documents/matryoshka/unet",
5
  "act_fn": "silu",
@@ -52,7 +52,7 @@
52
  "resnet_skip_time_act": false,
53
  "resnet_time_scale_shift": "scale_shift",
54
  "reverse_transformer_layers_per_block": null,
55
- "sample_size": null,
56
  "time_cond_proj_dim": null,
57
  "time_embedding_act_fn": null,
58
  "time_embedding_dim": null,
 
1
  {
2
+ "_class_name": "MatryoshkaUNet2DConditionModel",
3
  "_diffusers_version": "0.31.0.dev0",
4
  "_name_or_path": "/home/cosmos/Documents/matryoshka/unet",
5
  "act_fn": "silu",
 
52
  "resnet_skip_time_act": false,
53
  "resnet_time_scale_shift": "scale_shift",
54
  "reverse_transformer_layers_per_block": null,
55
+ "sample_size": 64,
56
  "time_cond_proj_dim": null,
57
  "time_embedding_act_fn": null,
58
  "time_embedding_dim": null,