diff options
| author | William Moorehouse <moorehousew@gmail.com> | 2022-10-09 04:58:40 -0400 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-09 14:49:15 +0300 |
| commit | d6d10a37bfd21568e74efb46137f906da96d5fdb (patch) | |
| tree | 2354d9dfeb9d75393b6333297a3a504e5bb4b1cf /modules/sd_models.py | |
| parent | 542a3d3a4a00c1383fbdaf938ceefef87cf834bb (diff) | |
Added extended model details to infotext
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index d0c74dd8..3fa42329 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -4,7 +4,7 @@ import sys from collections import namedtuple
import torch
from omegaconf import OmegaConf
-
+from pathlib import Path
from ldm.util import instantiate_from_config
@@ -158,6 +158,7 @@ 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
|