Instructions to use flax/waifu-diffusion with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use flax/waifu-diffusion with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("flax/waifu-diffusion", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Commit History
-
Fix deprecation warning by changing `CLIPFeatureExtractor` to `CLIPImageProcessor`.
Hey flax 👋, Your model repository seems to contain logic to load a feature extractor that is deprecated, which you should notice by seeing the warning: ``` transformers/models/clip/feature_extraction_clip.py:28: FutureWarning: The class CLIPFeatureExtractor is deprecated and will be removed in version 5 of Transformers. Please use CLIPImageProcessor instead. warnings.warn( ``` when running `pipe = DiffusionPipeline.from_pretrained(flax/waifu-diffusion)`.This PR makes sure that the warning does not show anymore by replacing `CLIPFeatureExtractor` with `CLIPImageProcessor`. This will certainly not change or break your checkpoint, but onlymake sure that everything is up to date. Best, the 🧨 Diffusers team.a349825
-
Add `scale_factor` to vae config. (#2)
- Add `scale_factor` to vae config. (c19e763cf12193737fb3c18a5ef547364df38821) Co-authored-by: Suraj Patil <valhalla@users.noreply.huggingface.co>1e3520f
-
Add `clip_sample=False` to scheduler to make model compatible with DDIM. (#1)
- Add `clip_sample=False` to scheduler to make model compatible with DDIM. (e39e574d1b5bdc3949d1d7d62016a4a0ec45c33b) Co-authored-by: Patrick von Platen <patrickvonplaten@users.noreply.huggingface.co>b32d031
-
Update unet/config.json
3e43301
-
tags
23b9184 -
tags
de3d1f0 -
flax
e8702f3 -
initial commit
bdd59b8