From 497645b2f8df4e64e7234f46ee446ee1d8d80efd Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 12:56:17 -0800 Subject: [PATCH] Update interrogate.py --- modules/interrogate.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/modules/interrogate.py b/modules/interrogate.py index 43365a83..78059fe2 100644 --- a/modules/interrogate.py +++ b/modules/interrogate.py @@ -117,6 +117,9 @@ class InterrogateModels: def load_clip_model(self): import clip + import clip.model + + clip.model.LayerNorm = torch.nn.LayerNorm model, preprocess = clip.load(clip_model_name, device="cpu", download_root=shared.cmd_opts.clip_models_path) model.eval()