mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-06 11:32:31 +08:00
Update cfz_vae_loader.py
This commit is contained in:
parent
79f734309b
commit
5f05fe3e1d
@ -45,7 +45,7 @@ class CFZVAELoader:
|
|||||||
# Load the VAE with patched precision functions
|
# Load the VAE with patched precision functions
|
||||||
vae_loader = VAELoader()
|
vae_loader = VAELoader()
|
||||||
vae = vae_loader.load_vae(vae_name)[0]
|
vae = vae_loader.load_vae(vae_name)[0]
|
||||||
print(f"CFZ VAE: Loaded with forced precision {precision}")
|
print(f"[CFZ VAE] VAE loaded with precision {precision}")
|
||||||
return (vae,)
|
return (vae,)
|
||||||
finally:
|
finally:
|
||||||
# Restore original functions
|
# Restore original functions
|
||||||
@ -59,4 +59,5 @@ NODE_CLASS_MAPPINGS = {
|
|||||||
|
|
||||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||||
"CFZVAELoader": "CFZ VAE Loader"
|
"CFZVAELoader": "CFZ VAE Loader"
|
||||||
}
|
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user