tolgacangoz
/

matryoshka-diffusion-models

tolgacangoz commited on Oct 6, 2024

Commit

f16b5c7

verified ·

1 Parent(s): d3d0ea5

Update unet/nesting_level_0/config.json

Files changed (1) hide show

unet/nesting_level_0/config.json CHANGED Viewed

@@ -68,5 +68,6 @@
     "UpBlock2D"
   ],
   "upcast_attention": false,
-  "use_linear_projection": false
 }

     "UpBlock2D"
   ],
   "upcast_attention": false,
+  "use_linear_projection": false,
+  "torch_dtype": "float32",
 }