huggingface / huggingface/diffusion-models-class

A parameter needs to be specified in unit 1 - 01_introduction_to_diffusers.ipynb

Open
#92 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
4.4k
Forks
492
PR merge metrics
No merged PRs in 30d

Description

In this notebook, originally there is a line of code:
`image_pipe.save_pretrained("my_pipeline")`
, which will generate a saved model as 'diffusion_pytorch_model.safetensors'. However, it will cause errors in the following line
`pipeline = DDPMPipeline.from_pretrained(hub_model_id)`

After investigation, we have to specify one more parameter when saving model as follows,
`image_pipe.save_pretrained("my_pipeline", safe_serialization=False)`

Then, it should be work when loading our own trained model.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.