Skip to content

GitLab

Commit 587f89fe authored by comfyanonymous's avatar comfyanonymous
Browse files

Enable safe loading for upscale models.

parent acff543d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy_extras/nodes_upscale_model.py
View file @ 587f89fe
...@@ -17,7 +17,7 @@ class UpscaleModelLoader: ...@@ -17,7 +17,7 @@ class UpscaleModelLoader:
def load_model(self, model_name): def load_model(self, model_name):
model_path = folder_paths.get_full_path("upscale_models", model_name) model_path = folder_paths.get_full_path("upscale_models", model_name)
sd = comfy.utils.load_torch_file(model_path) sd = comfy.utils.load_torch_file(model_path, safe_load=True)
out = model_loading.load_state_dict(sd).eval() out = model_loading.load_state_dict(sd).eval()
return (out, ) return (out, )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment