hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

change up #2056 to make it work how i want it to plus make xy plot write correct values to images

Browse Source
This commit is contained in:
AUTOMATIC
2022-10-09 14:57:48 +03:00
parent 594cbfd8fb
commit e6e8cabe0c
3 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models.py
View File

@@ -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