Update unet/config.json
Browse files- unet/config.json +1 -2
unet/config.json
CHANGED
@@ -21,6 +21,5 @@
|
|
21 |
"num_head_channels": 64,
|
22 |
"num_res_blocks": 2,
|
23 |
"out_channels": 4,
|
24 |
-
"transformer_depth": 1
|
25 |
-
"use_checkpoint": false
|
26 |
}
|
|
|
21 |
"num_head_channels": 64,
|
22 |
"num_res_blocks": 2,
|
23 |
"out_channels": 4,
|
24 |
+
"transformer_depth": 1
|
|
|
25 |
}
|