Added image resizing related nodes and a single image prediction node

2025-03-08 03:38:52 -08:00
parent a2fc03f41f
commit 5fee9ffb60
32 changed files with 1007 additions and 13 deletions
@@ -36,6 +36,6 @@ class PtLoadModel:

    def f(self, model:Any, model_path: str):
        model_full_path = get_model_full_path(model_path)
-        sd = torch.load(model_full_path)
+        sd = torch.load(model_full_path, weights_only=True)
        model.load_state_dict(sd)
        return (model,)