correctly cache horde-safety models
This commit is contained in:
parent
0d1f65a010
commit
415405c2b2
|
@ -39,7 +39,7 @@ class EditSafetyStage(BaseStage):
|
||||||
deep_danbooru_model,
|
deep_danbooru_model,
|
||||||
)
|
)
|
||||||
|
|
||||||
server.cache.set(ModelTypes.safety, cache_key)
|
server.cache.set(ModelTypes.safety, cache_key, nsfw_checker)
|
||||||
|
|
||||||
return nsfw_checker
|
return nsfw_checker
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue