Commit 47f4fecf authored by kurumuz's avatar kurumuz

fix agtain

parent d7e64795
......@@ -104,7 +104,7 @@ def sanitize_stable_diffusion(request, config):
random.setstate(state)
if request.module is not None:
if request.module not in config.model.premodules or request.module != "vanilla":
if request.module not in config.model.premodules and request.module != "vanilla":
return False, "module should be one of: " + ", ".join(config.model.premodules)
if request.image is not None:
......
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