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,
|