refactor: code cleanup
This commit is contained in:
parent
4a3aac09a3
commit
af33e930d3
|
|
@ -71,9 +71,9 @@ def refresh_base_model(name, vae_name=None):
|
|||
if model_base.filename == filename and model_base.vae_filename == vae_filename:
|
||||
return
|
||||
|
||||
model_base = core.StableDiffusionModel(vae_filename=vae_filename)
|
||||
model_base = core.load_model(filename, vae_filename)
|
||||
print(f'Base model loaded: {model_base.filename}')
|
||||
print(f'VAE loaded: {model_base.vae_filename}')
|
||||
return
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue