3v324v23 commited on
Commit
8847ac0
1 Parent(s): d32e573

200 steps trained @ 150 batch size on 1080p and portrait images

Browse files
Files changed (1) hide show
  1. model_index.json +2 -2
model_index.json CHANGED
@@ -23,8 +23,8 @@
23
  "CLIPTokenizer"
24
  ],
25
  "unet": [
26
- "torch",
27
- "OptimizedModule"
28
  ],
29
  "vae": [
30
  "diffusers",
 
23
  "CLIPTokenizer"
24
  ],
25
  "unet": [
26
+ "diffusers",
27
+ "UNet2DConditionModel"
28
  ],
29
  "vae": [
30
  "diffusers",