ashawkey commited on
Commit
78eec61
1 Parent(s): 7521d91

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +1 -2
unet/config.json CHANGED
@@ -20,6 +20,5 @@
20
  "num_head_channels": 64,
21
  "num_res_blocks": 2,
22
  "out_channels": 4,
23
- "transformer_depth": 1,
24
- "use_checkpoint": false
25
  }
 
20
  "num_head_channels": 64,
21
  "num_res_blocks": 2,
22
  "out_channels": 4,
23
+ "transformer_depth": 1
 
24
  }