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