add partial .safetensors support

[ProducerMatt]

Allows loading of ST and saving as ST. To verify, merge two small models that don't need pruning.

Can't be merged yet as there's no support in prune and the other external scripts yet. I got stuck on the details regarding ema_keys. When I have time I'll try to take a shot at it.

Should I make a file for shared functions, i.e. model loading/saving funcs for ckpt+safetensors?

[diStyApps]

Hey great work,
Thank you for your work on adding support for SafeTensors.

Can't be merged yet as there's no support in prune and the other external scripts yet. I got stuck on the details regarding ema_keys. When I have time I'll try to take a shot at it.

i did merged .safetensors and .ckpt your PR successfully.

Important to note the "Use FP16" is how the model merges, and how its exactly saves the model.

and you can save .safetensors to fp16 by adding this line to make_safetensors:

if make_safetensors:
    theta_0 = {name: tensor.half() for name, tensor in theta_0.items()}

    safetensors.torch.save_file(theta_0, output_file)

The app originally functioned as a direct GUI for the source, but some of its behavior was not correct.

Planned updates include:

Adding the ability to save models as FP16.
Adding a new button to convert models to FP16.
Adding support for SafeTensors (which has already been implemented by you).

it appears that the current PR introduces a bug that prevents models from being saved in the .ckpt format and makes .safetensors the default format even if it is not selected.

Should I make a file for shared functions, i.e. model loading/saving funcs for ckpt+safetensors?

sounds like a good idea.