From 415405c2b2c3a018a3f66fee8949a5924f6a110c Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 2 Mar 2024 13:57:35 -0600 Subject: [PATCH] correctly cache horde-safety models --- api/onnx_web/chain/edit_safety.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/chain/edit_safety.py b/api/onnx_web/chain/edit_safety.py index afcb7fb6..d4635b2e 100644 --- a/api/onnx_web/chain/edit_safety.py +++ b/api/onnx_web/chain/edit_safety.py @@ -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