anton-l HF staff commited on
Commit
24a4f79
·
1 Parent(s): f2c4981

update pipeline configs

Browse files
Files changed (1) hide show
  1. text_unet/config.json +1 -0
text_unet/config.json CHANGED
@@ -22,6 +22,7 @@
22
  "num_res_blocks": 3,
23
  "out_channels": 6,
24
  "resblock_updown": true,
 
25
  "use_checkpoint": false,
26
  "use_fp16": false,
27
  "use_scale_shift_norm": true
 
22
  "num_res_blocks": 3,
23
  "out_channels": 6,
24
  "resblock_updown": true,
25
+ "transformer_dim": 512,
26
  "use_checkpoint": false,
27
  "use_fp16": false,
28
  "use_scale_shift_norm": true