This commit is contained in:
Haoming 2025-10-29 11:38:33 +08:00
parent f9f0954024
commit f9c8bd2b85
2 changed files with 1 additions and 31 deletions

View File

@ -170,11 +170,8 @@ def configure_sigint_handler():
def configure_opts_onchange():
from modules import shared, sd_vae, ui_tempdir
from modules.call_queue import wrap_queued_call
from modules_forge import main_thread
from modules import shared, ui_tempdir
shared.opts.onchange("sd_vae_overrides_per_model_preferences", wrap_queued_call(lambda: main_thread.run_and_wait_result(sd_vae.reload_vae_weights)), call=False)
shared.opts.onchange("temp_dir", ui_tempdir.on_tmpdir_changed)
shared.opts.onchange("gradio_theme", shared.reload_gradio_theme)
startup_timer.record("opts onchange")

View File

@ -6,10 +6,8 @@ from dataclasses import dataclass
from modules import extra_networks, hashes, paths, sd_models, shared
vae_path = os.path.abspath(os.path.join(paths.models_path, "VAE"))
vae_ignore_keys = {"model_ema.decay", "model_ema.num_updates"}
vae_dict = {}
base_vae = None
loaded_vae_file = None
checkpoint_info = None
@ -55,7 +53,6 @@ def restore_base_vae(model):
global loaded_vae_file
if base_vae is not None and checkpoint_info == model.sd_checkpoint_info:
print("Restoring base VAE")
_load_vae_dict(model, base_vae)
loaded_vae_file = None
delete_base_vae()
@ -166,27 +163,3 @@ def resolve_vae(checkpoint_file) -> VaeResolution:
res = resolve_vae_from_setting()
return res
def load_vae_dict(filename, map_location):
pass
def load_vae(model, vae_file=None, vae_source="from unknown source"):
pass
# don't call this from outside
def _load_vae_dict(model, vae_dict_1):
pass
def clear_loaded_vae():
pass
unspecified = object()
def reload_vae_weights(sd_model=None, vae_file=unspecified):
pass