English
Generated from Trainer
Inference Endpoints
Michael Brunzel commited on
Commit
9912bfa
1 Parent(s): 2d660d0

Exclude unnecessary params

Browse files
Files changed (1) hide show
  1. 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,