"...resnet50_tensorflow.git" did not exist on "3635527dc66cdfe7270e5b3086858db7307df8a3"
  • Matthieu Bizien's avatar
    [Proposal] Support saving to safetensors (#1494) · ae368e42
    Matthieu Bizien authored
    * Add parameter safe_serialization to DiffusionPipeline.save_pretrained
    
    * Add option safe_serialization on ModelMixin.save_pretrained
    
    * Add test test_save_safe_serialization
    
    * Black
    
    * Re-trigger the CI
    
    * Fix doc-builder
    
    * Validate files are saved as safetensor in test_save_safe_serialization
    ae368e42
test_pipelines.py 34.9 KB