Spaces:
Runtime error
Runtime error
Update ominicontrol.py
Browse files- ominicontrol.py +2 -1
ominicontrol.py
CHANGED
|
@@ -63,7 +63,7 @@ def generate_image(
|
|
| 63 |
"The Simpsons": "simpsons",
|
| 64 |
"Snoopy": "snoopy",
|
| 65 |
}[style]
|
| 66 |
-
|
| 67 |
|
| 68 |
factor = 512 / max(image.size)
|
| 69 |
image = resize(
|
|
@@ -132,6 +132,7 @@ def generate_image(
|
|
| 132 |
"image_guidance_scale": image_guidance,
|
| 133 |
"seed": seed,
|
| 134 |
"steps": steps,
|
|
|
|
| 135 |
"width": width,
|
| 136 |
"height": height,
|
| 137 |
}
|
|
|
|
| 63 |
"The Simpsons": "simpsons",
|
| 64 |
"Snoopy": "snoopy",
|
| 65 |
}[style]
|
| 66 |
+
pipe.set_adapters(activate_adapter_name)
|
| 67 |
|
| 68 |
factor = 512 / max(image.size)
|
| 69 |
image = resize(
|
|
|
|
| 132 |
"image_guidance_scale": image_guidance,
|
| 133 |
"seed": seed,
|
| 134 |
"steps": steps,
|
| 135 |
+
"style": style,
|
| 136 |
"width": width,
|
| 137 |
"height": height,
|
| 138 |
}
|