resolve VAE files within model path
This commit is contained in:
parent
2c66cc58c5
commit
aa6dc334d7
|
@ -1311,6 +1311,7 @@ def extract_checkpoint(
|
||||||
if vae_file is None:
|
if vae_file is None:
|
||||||
converted_vae_checkpoint = convert_ldm_vae_checkpoint(checkpoint, vae_config)
|
converted_vae_checkpoint = convert_ldm_vae_checkpoint(checkpoint, vae_config)
|
||||||
else:
|
else:
|
||||||
|
vae_file = os.path.join(ctx.model_path, vae_file)
|
||||||
vae_checkpoint = safetensors.torch.load_file(vae_file, device="cpu")
|
vae_checkpoint = safetensors.torch.load_file(vae_file, device="cpu")
|
||||||
converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_checkpoint, vae_config)
|
converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_checkpoint, vae_config)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue