Commit 1a6751df authored by novelailab's avatar novelailab

should work

parent c355eef8
...@@ -69,7 +69,7 @@ def sanitize_input(config, request): ...@@ -69,7 +69,7 @@ def sanitize_input(config, request):
Sanitize the input data and set defaults Sanitize the input data and set defaults
""" """
request = DotMap(request) request = DotMap(request)
default, forced_default = defaults[config.model] default, forced_default = defaults[config.model_name]
for k, v in default.items(): for k, v in default.items():
if k not in request: if k not in request:
request[k] = v request[k] = v
...@@ -77,9 +77,9 @@ def sanitize_input(config, request): ...@@ -77,9 +77,9 @@ def sanitize_input(config, request):
for k, v in forced_default.items(): for k, v in forced_default.items():
request[k] = v request[k] = v
if config.model == 'stable-diffusion': if config.model_name == 'stable-diffusion':
return sanitize_stable_diffusion(request) return sanitize_stable_diffusion(request)
elif config.model == 'dalle-mini': elif config.model_name == 'dalle-mini':
return sanitize_dalle_mini(request) return sanitize_dalle_mini(request)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment