Commit
•
0bb64a4
1
Parent(s):
690f19f
fix call kwargs
Browse files- app.py +1 -1
- config_store.py +1 -1
app.py
CHANGED
@@ -66,7 +66,7 @@ def parse_configs(inputs):
|
|
66 |
|
67 |
for key in configs.keys():
|
68 |
for k, v in configs[key].items():
|
69 |
-
if k in ["input_shapes", "generate_kwargs", "numactl_kwargs"]:
|
70 |
configs[key][k] = eval(v)
|
71 |
|
72 |
configs["process"] = ProcessConfig(**configs.pop("process"))
|
|
|
66 |
|
67 |
for key in configs.keys():
|
68 |
for k, v in configs[key].items():
|
69 |
+
if k in ["input_shapes", "generate_kwargs", "call_kwargs", "numactl_kwargs"]:
|
70 |
configs[key][k] = eval(v)
|
71 |
|
72 |
configs["process"] = ProcessConfig(**configs.pop("process"))
|
config_store.py
CHANGED
@@ -64,7 +64,7 @@ def get_inference_config():
|
|
64 |
),
|
65 |
"inference.call_kwargs": gr.Textbox(
|
66 |
label="inference.call_kwargs",
|
67 |
-
value="{'num_inference_steps':
|
68 |
info="Additional python dict of kwargs to pass to the __call__ method",
|
69 |
),
|
70 |
}
|
|
|
64 |
),
|
65 |
"inference.call_kwargs": gr.Textbox(
|
66 |
label="inference.call_kwargs",
|
67 |
+
value="{'num_inference_steps': 5}",
|
68 |
info="Additional python dict of kwargs to pass to the __call__ method",
|
69 |
),
|
70 |
}
|