eplace torch.save with another method. Can be configured with the environment variable
DIFFUSERS_SAVE_MODE.   safe_serialization (bool, optional, defaults to True) —
Whether to save the model using safetensors or the traditional PyTorch way with pickle.   variant (str, optional) —
If specified, weights are saved in the format pytorch_model.<variant>.bin.   push_to_hub (bool, optional, defaults to False) —
Whether or not to push your model to th