2kpr / ComfyUI-UltraPixel

GNU Affero General Public License v3.0
215 stars 11 forks source link

The "safetensors" files are NOT safetensors files! They are potentially dangerous pytorch saves, just with the ".safetensors" extension! #28

Open ClownsharkBatwing opened 3 months ago

ClownsharkBatwing commented 3 months ago

The "safetensors" files provided for UltraPixel (ultrapixel_t2i.safetensors, lora_cat.safetensors) are not actually safetensors files at all. They are pytorch saves - potentially dangerous binaries - that misleadingly are named with the ".safetensors" extension. I converted both of them to legitimate safetensors files, and submitted a PR to the HF repo:

https://huggingface.co/ClownsharkBatwing/ultrapixel_convert/blob/main/ultrapixel_t2i.safetensors https://huggingface.co/ClownsharkBatwing/lora_cat_convert/blob/main/lora_cat.safetensors https://huggingface.co/roubaofeipi/UltraPixel/discussions/1

I submitted a PR for this GitHub repo that implements the ability to load these safetensor versions, in place of the pytorch saves. The changes made were minimal, and limited only to ultrapixel.py so if you wish to use real safetensors files while we wait for review, that's the only file you'll have to replace. https://github.com/2kpr/ComfyUI-UltraPixel/pull/27