apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

correctly cache horde-safety models

This commit is contained in:
Sean Sube 2024-03-02 13:57:35 -06:00
parent 0d1f65a010
commit 415405c2b2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/edit_safety.py
View File

@ -39,7 +39,7 @@ class EditSafetyStage(BaseStage):
deep_danbooru_model,
)
server.cache.set(ModelTypes.safety, cache_key)
server.cache.set(ModelTypes.safety, cache_key, nsfw_checker)
return nsfw_checker