Just regular .safetensors files pretty please?

#1
by kulminaator - opened

The current files here are .pt and these break with new pytorch & comfyui.

// logs from comfy

# ComfyUI Error Report
## Error Details
- **Node ID:** 170
- **Node Type:** UNETLoaderMultiGPU
- **Exception Type:** _pickle.UnpicklingError
- **Exception Message:** _pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, do those steps only if you trust the source of the checkpoint. 
    (1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source.
    (2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message.
    WeightsUnpickler error: Unsupported global: GLOBAL torch.torch_version.TorchVersion was not an allowed global by default. Please use `torch.serialization.add_safe_globals([torch.torch_version.TorchVersion])` or the `torch.serialization.safe_globals([torch.torch_version.TorchVersion])` context manager to allowlist this global if you trust this class/function.

Sign up or log in to comment