diff options
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
|