Spaces:
Runtime error
Runtime error
hsuwill000
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{tae
|
|
42 |
"""
|
43 |
print(pipe.scheduler.compatibles)
|
44 |
|
45 |
-
pipe.reshape(batch_size=batch_size, height=HIGH, width=WIDTH, num_images_per_prompt=
|
46 |
|
47 |
pipe.compile()
|
48 |
|
@@ -55,9 +55,10 @@ def infer(prompt, negative_prompt, num_inference_steps=1):
|
|
55 |
negative_prompt=negative_prompt,
|
56 |
width=WIDTH,
|
57 |
height=HIGH,
|
58 |
-
guidance_scale=
|
59 |
num_inference_steps=num_inference_steps,
|
60 |
num_images_per_prompt=1,
|
|
|
61 |
).images[0]
|
62 |
|
63 |
return image
|
|
|
42 |
"""
|
43 |
print(pipe.scheduler.compatibles)
|
44 |
|
45 |
+
pipe.reshape(batch_size=batch_size, height=HIGH, width=WIDTH, num_images_per_prompt=2)
|
46 |
|
47 |
pipe.compile()
|
48 |
|
|
|
55 |
negative_prompt=negative_prompt,
|
56 |
width=WIDTH,
|
57 |
height=HIGH,
|
58 |
+
guidance_scale=0.0,
|
59 |
num_inference_steps=num_inference_steps,
|
60 |
num_images_per_prompt=1,
|
61 |
+
strength=0.5,
|
62 |
).images[0]
|
63 |
|
64 |
return image
|