Instructions to use flax/Tron-Legacy-diffusion with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use flax/Tron-Legacy-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/Tron-Legacy-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/Tron-Legacy-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.c6629b8
-
Add `scale_factor` to vae config. (#2)
- Add `scale_factor` to vae config. (726f5d56c10695d1ef900c459a3eb1c33bfc4c56) Co-authored-by: Suraj Patil <valhalla@users.noreply.huggingface.co>31a4acf
-
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. (a0ce3ffe24c6b8bfc59018c9fc1915f8d129459b) Co-authored-by: Patrick von Platen <patrickvonplaten@users.noreply.huggingface.co>1754454
-
Update unet/config.json
d213651
-
tags
e33b465
-
flax
c23a6dd -
initial commit
00f8635