diff options
| author | MalumaDev <piano.lu92@gmail.com> | 2022-10-16 15:53:56 +0000 |
|---|---|---|
| committer | MalumaDev <piano.lu92@gmail.com> | 2022-10-16 15:53:56 +0000 |
| commit | 9324cdaa3199d65c182858785dd1eca42b192b8e (patch) | |
| tree | 47dc89324b5b4528a1744e269d82a5106bfa4e60 /modules/sd_models.py | |
| parent | e4f8b5f00dd33b7547cc6b76fbed26bb83b37a64 (diff) | |
| download | stable-diffusion-webui-gfx803-9324cdaa3199d65c182858785dd1eca42b192b8e.tar.gz stable-diffusion-webui-gfx803-9324cdaa3199d65c182858785dd1eca42b192b8e.tar.bz2 stable-diffusion-webui-gfx803-9324cdaa3199d65c182858785dd1eca42b192b8e.zip | |
ui fix, re organization of the code
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 3aa21ec1..8e4ee435 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -20,7 +20,7 @@ checkpoints_loaded = collections.OrderedDict() try:
# this silences the annoying "Some weights of the model checkpoint were not used when initializing..." message at start.
- from transformers import logging
+ from transformers import logging, CLIPModel
logging.set_verbosity_error()
except Exception:
@@ -196,6 +196,9 @@ def load_model(): sd_hijack.model_hijack.hijack(sd_model)
+ if shared.clip_model is None or shared.clip_model.transformer.name_or_path != sd_model.cond_stage_model.wrapped.transformer.name_or_path:
+ shared.clip_model = CLIPModel.from_pretrained(sd_model.cond_stage_model.wrapped.transformer.name_or_path)
+
sd_model.eval()
print(f"Model loaded.")
|