diffusers
diffusers copied to clipboard
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
TODO: - [ ] Run an actual benchmark testing that it's faster - [ ] Maybe `Linear` doesn't make sense as running `Conv1d` would remove quite a few reshape/contiguous etc...
Added sampling script that supports both DDPM and DDIM sampling for unconditional training example
This project works great. Is there a code for pre-training stable diffusion?
### Describe the bug When using ONNX with the custom pipeline `lpw_stable_diffusion_onnx` it raises a "unsupported operand" error. It's the same error as this other bug #967 ### Reproduction ```python...
train_dreambooth.py adapted to work with the inpaint model, generating random masks during the training
For issue #320 and #544
**Context** Currently [`OnnxStableDiffusionPipeline`](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py#LL34C7-L34C34) perform unnecessary tensor casting between torch and numpy. The downsides of that are: * The pipeline code is harder to maintain: we have to keep track of...
### Describe the bug I'm not able to run the Flax pipeline example of [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4) on Colab with GPU enabled. I get this error ``` ValueError: pmap got inconsistent sizes...
`.from_pretrained()` can work in offline mode by loading from the cache, but we lack a method to explicitly populate that cache. I'd like something along the lines of `.cache_pretrained_for_offline()` that...
It is a common use case to want to know exactly where the pipeline was downloaded. This PR adds a simply kwarg `return_cached_folder` to see this. It could help with...