diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 14:57:48 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 14:57:48 +0300 |
| commit | e6e8cabe0c9c335e0d72345602c069b198558b53 (patch) | |
| tree | 3c0b58a0792336be7b6886ab44e69d9e574837c5 /modules/sd_models.py | |
| parent | 594cbfd8fbe4078b43ceccf01509eeef3d6790c6 (diff) | |
change up #2056 to make it work how i want it to plus make xy plot write correct values to images
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 3fa42329..e63d3c29 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -4,7 +4,6 @@ import sys from collections import namedtuple
import torch
from omegaconf import OmegaConf
-from pathlib import Path
from ldm.util import instantiate_from_config
@@ -158,7 +157,6 @@ def load_model_weights(model, checkpoint_info): vae_dict = {k: v for k, v in vae_ckpt["state_dict"].items() if k[0:4] != "loss"}
model.first_stage_model.load_state_dict(vae_dict)
- model.sd_model_vae_name = Path(vae_file).stem
model.sd_model_hash = sd_model_hash
model.sd_model_checkpoint = checkpoint_file
|