patrickvonplaten commited on
Commit
b57731e
·
1 Parent(s): f4b06bc

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +28 -1
config.json CHANGED
@@ -42,5 +42,32 @@
42
  "resamp_with_conv": true,
43
  "resblock_type": "biggan",
44
  "scale_by_sigma": true,
45
- "skip_rescale": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  }
 
42
  "resamp_with_conv": true,
43
  "resblock_type": "biggan",
44
  "scale_by_sigma": true,
45
+ "skip_rescale": true,
46
+ "block_channels": [128, 128, 256, 256, 256, 256, 256],
47
+ "in_channels": 3,
48
+ "out_channels": 3,
49
+ "time_embedding_type": "fourier",
50
+ "resnet_eps": 1e-6,
51
+ "mid_block_scale_factor": 1.41421356237,
52
+ "resnet_num_groups": null,
53
+ "down_blocks": [
54
+ "UNetResSkipDownBlock2D",
55
+ "UNetResSkipDownBlock2D",
56
+ "UNetResSkipDownBlock2D",
57
+ "UNetResSkipDownBlock2D",
58
+ "UNetResAttnSkipDownBlock2D",
59
+ "UNetResSkipDownBlock2D",
60
+ "UNetResSkipDownBlock2D"
61
+ ],
62
+ "up_blocks": [
63
+ "UNetResSkipUpBlock2D",
64
+ "UNetResSkipUpBlock2D",
65
+ "UNetResAttnSkipUpBlock2D",
66
+ "UNetResSkipUpBlock2D",
67
+ "UNetResSkipUpBlock2D",
68
+ "UNetResSkipUpBlock2D",
69
+ "UNetResSkipUpBlock2D"
70
+ ],
71
+ "center_input_sample": true,
72
+ "num_head_channels": null
73
  }