karimbenharrak
commited on
Commit
•
bd0e4b7
1
Parent(s):
81e129b
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -96,7 +96,7 @@ class EndpointHandler():
|
|
96 |
self.pipe.enable_xformers_memory_efficient_attention()
|
97 |
|
98 |
# run inference pipeline
|
99 |
-
out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image
|
100 |
|
101 |
print("1st pipeline part successful!")
|
102 |
|
|
|
96 |
self.pipe.enable_xformers_memory_efficient_attention()
|
97 |
|
98 |
# run inference pipeline
|
99 |
+
out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image)
|
100 |
|
101 |
print("1st pipeline part successful!")
|
102 |
|