zhiweili
commited on
Commit
·
7a58dc4
1
Parent(s):
8772d2d
add prompt
Browse files- app_tensorrt.py +2 -1
app_tensorrt.py
CHANGED
|
@@ -52,7 +52,8 @@ base_pipe.unet = compiledModel
|
|
| 52 |
|
| 53 |
init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img.png")
|
| 54 |
generated_image = base_pipe(
|
| 55 |
-
init_image,
|
|
|
|
| 56 |
num_inference_steps=5,
|
| 57 |
).images[0]
|
| 58 |
|
|
|
|
| 52 |
|
| 53 |
init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img.png")
|
| 54 |
generated_image = base_pipe(
|
| 55 |
+
image=init_image,
|
| 56 |
+
prompt="A white cat",
|
| 57 |
num_inference_steps=5,
|
| 58 |
).images[0]
|
| 59 |
|