Fix a bug where the vae wouldn't be unloaded in webui

This commit is contained in:
cmdr2 2024-10-09 13:41:56 +05:30
parent 3327244da2
commit d6adb17746

View File

@ -137,7 +137,9 @@ def load_model(context, model_type, **kwargs):
def unload_model(context, model_type, **kwargs):
pass
if model_type == "vae":
context.model_paths[model_type] = None
load_model(context, model_type)
def generate_images(