We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ec53008 commit e8aa61bCopy full SHA for e8aa61b
src/diffusers/models/modeling_utils.py
@@ -714,7 +714,10 @@ def save_pretrained(
714
if safe_serialization:
715
# At some point we will need to deal better with save_function (used for TPU and other distributed
716
# joyfulness), but for now this enough.
717
- safetensors.torch.save_file(shard, filepath, metadata={"format": "pt"})
+ try:
718
+ safetensors.torch.save_file(shard, filepath, metadata={"format": "pt"})
719
+ except RuntimeError:
720
+ safetensors.torch.save_model(model_to_save, filepath, metadata={"format": "pt"})
721
else:
722
torch.save(shard, filepath)
723
0 commit comments