lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-28 02:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update interrogate.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 12:56:17 -08:00
parent 63811455ef
commit 497645b2f8
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/interrogate.py
View File

@@ -117,6 +117,9 @@ class InterrogateModels:
def load_clip_model(self): def load_clip_model(self):
import clip 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, preprocess = clip.load(clip_model_name, device="cpu", download_root=shared.cmd_opts.clip_models_path)
model.eval() model.eval()