bug fix
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -76,7 +76,7 @@ class EndpointHandler:
|
|
| 76 |
image=image,
|
| 77 |
)
|
| 78 |
else:
|
| 79 |
-
out = self.
|
| 80 |
prompt,
|
| 81 |
num_inference_steps=num_inference_steps,
|
| 82 |
guidance_scale=guidance_scale,
|
|
|
|
| 76 |
image=image,
|
| 77 |
)
|
| 78 |
else:
|
| 79 |
+
out = self.base(
|
| 80 |
prompt,
|
| 81 |
num_inference_steps=num_inference_steps,
|
| 82 |
guidance_scale=guidance_scale,
|