Enable safe loading for upscale models.

587f89fe · comfyanonymous · acff543d · 587f89fe
Commit 587f89fe authored May 14, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy_extras/nodes_upscale_model.py comfy_extras/nodes_upscale_model.py +1 -1

No files found.
--- a/comfy_extras/nodes_upscale_model.py
+++ b/comfy_extras/nodes_upscale_model.py
@@ -17,7 +17,7 @@ class UpscaleModelLoader:

    def load_model(self, 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()
        return (out, )