patrickvonplaten commited on
Commit
8236408
·
1 Parent(s): 45bcf86

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +29 -1
config.json CHANGED
@@ -42,5 +42,33 @@
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": [16, 32, 64, 128, 256, 512, 512, 512],
47
+ "in_channels": 3,
48
+ "num_res_blocks": 1,
49
+ "out_channels": 3,
50
+ "time_embedding_type": "fourier",
51
+ "resnet_eps": 1e-6,
52
+ "mid_block_scale_factor": 1.41421356237,
53
+ "resnet_num_groups": null,
54
+ "down_blocks": [
55
+ "UNetResSkipDownBlock2D",
56
+ "UNetResAttnSkipDownBlock2D",
57
+ "UNetResSkipDownBlock2D",
58
+ "UNetResSkipDownBlock2D",
59
+ "UNetResSkipDownBlock2D",
60
+ "UNetResSkipDownBlock2D",
61
+ "UNetResSkipDownBlock2D",
62
+ "UNetResSkipDownBlock2D"
63
+ ],
64
+ "up_blocks": [
65
+ "UNetResSkipUpBlock2D",
66
+ "UNetResSkipUpBlock2D",
67
+ "UNetResSkipUpBlock2D",
68
+ "UNetResSkipUpBlock2D",
69
+ "UNetResSkipUpBlock2D",
70
+ "UNetResSkipUpBlock2D",
71
+ "UNetResAttnSkipUpBlock2D",
72
+ "UNetResSkipUpBlock2D"
73
+ ]
74
  }