Update handler.py
Browse files- handler.py +2 -3
handler.py
CHANGED
|
@@ -59,7 +59,7 @@ class EndpointHandler():
|
|
| 59 |
|
| 60 |
#self.pipe2.enable_xformers_memory_efficient_attention()
|
| 61 |
|
| 62 |
-
image =
|
| 63 |
prompt=prompt,
|
| 64 |
image=image,
|
| 65 |
mask_image=mask_image,
|
|
@@ -70,10 +70,9 @@ class EndpointHandler():
|
|
| 70 |
output_type="latent", # let's keep in latent to save some VRAM
|
| 71 |
).images[0]
|
| 72 |
|
| 73 |
-
pipe = AutoPipelineForImage2Image.from_pipe(pipe)
|
| 74 |
#self.pipe3.enable_xformers_memory_efficient_attention()
|
| 75 |
|
| 76 |
-
image =
|
| 77 |
prompt=prompt,
|
| 78 |
image=image,
|
| 79 |
guidance_scale=8.0,
|
|
|
|
| 59 |
|
| 60 |
#self.pipe2.enable_xformers_memory_efficient_attention()
|
| 61 |
|
| 62 |
+
image = self.pipe2(
|
| 63 |
prompt=prompt,
|
| 64 |
image=image,
|
| 65 |
mask_image=mask_image,
|
|
|
|
| 70 |
output_type="latent", # let's keep in latent to save some VRAM
|
| 71 |
).images[0]
|
| 72 |
|
|
|
|
| 73 |
#self.pipe3.enable_xformers_memory_efficient_attention()
|
| 74 |
|
| 75 |
+
image = self.pipe3(
|
| 76 |
prompt=prompt,
|
| 77 |
image=image,
|
| 78 |
guidance_scale=8.0,
|