200 steps trained @ 150 batch size on 1080p and portrait images
Browse files- model_index.json +2 -2
model_index.json
CHANGED
@@ -23,8 +23,8 @@
|
|
23 |
"CLIPTokenizer"
|
24 |
],
|
25 |
"unet": [
|
26 |
-
"
|
27 |
-
"
|
28 |
],
|
29 |
"vae": [
|
30 |
"diffusers",
|
|
|
23 |
"CLIPTokenizer"
|
24 |
],
|
25 |
"unet": [
|
26 |
+
"diffusers",
|
27 |
+
"UNet2DConditionModel"
|
28 |
],
|
29 |
"vae": [
|
30 |
"diffusers",
|