Spaces:
Running
on
Zero
Running
on
Zero
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 |
}
|