parameter init_image test
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def infer(prompt, refer_image, height, width):
|
|
| 33 |
# num_inference_steps=num_inference_steps,
|
| 34 |
width=width,
|
| 35 |
height=height,
|
| 36 |
-
|
| 37 |
# generator=generator
|
| 38 |
).images[0]
|
| 39 |
|
|
|
|
| 33 |
# num_inference_steps=num_inference_steps,
|
| 34 |
width=width,
|
| 35 |
height=height,
|
| 36 |
+
init_image=refer_image
|
| 37 |
# generator=generator
|
| 38 |
).images[0]
|
| 39 |
|