xwen99
/

mar-vae-kl16

Model card Files Files and versions Community

xwen99 commited on Feb 11

Commit

53d104e

·

1 Parent(s): cc92c68

update model config

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -14,7 +14,7 @@
     "DownEncoderBlock2D",
     "DownEncoderBlock2D",
     "DownEncoderBlock2D",
-    "AttnDownBlock2D"
   ],
   "in_channels": 3,
   "latent_channels": 16,

     "DownEncoderBlock2D",
     "DownEncoderBlock2D",
     "DownEncoderBlock2D",
+    "AttnDownEncoderBlock2D"
   ],
   "in_channels": 3,
   "latent_channels": 16,