Michael Brunzel
commited on
Commit
·
9912bfa
1
Parent(s):
2d660d0
Exclude unnecessary params
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -49,7 +49,7 @@ class EndpointHandler:
|
|
| 49 |
if parameters is not None:
|
| 50 |
# Make endpoint compatible with hf client library
|
| 51 |
exclude_list = ["stop", "watermark", "details", "decoder_input_details"]
|
| 52 |
-
parameters = {name: val for name, val in parameters.items if name not in exclude_list}
|
| 53 |
|
| 54 |
outputs = self.pipe(
|
| 55 |
prompts,
|
|
|
|
| 49 |
if parameters is not None:
|
| 50 |
# Make endpoint compatible with hf client library
|
| 51 |
exclude_list = ["stop", "watermark", "details", "decoder_input_details"]
|
| 52 |
+
parameters = {name: val for name, val in parameters.items() if name not in exclude_list}
|
| 53 |
|
| 54 |
outputs = self.pipe(
|
| 55 |
prompts,
|