zhengqilin commited on
Commit
96fc329
·
1 Parent(s): 36cfce3
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -156,8 +156,8 @@ class EndpointHandler():
156
  "seed": -1,
157
  }
158
  if data["inputs"]:
159
- for field, v in args:
160
- if data["inputs"][field]:
161
  args[field] = data["inputs"][field]
162
  # if "prompt" in data["inputs"].keys():
163
  # prompt = data["inputs"]["prompt"]
 
156
  "seed": -1,
157
  }
158
  if data["inputs"]:
159
+ for field in args:
160
+ if field in data["inputs"].keys():
161
  args[field] = data["inputs"][field]
162
  # if "prompt" in data["inputs"].keys():
163
  # prompt = data["inputs"]["prompt"]