Instructions to use akshan-main/tiny-diffusion-gemma-modular-pipe with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use akshan-main/tiny-diffusion-gemma-modular-pipe with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("akshan-main/tiny-diffusion-gemma-modular-pipe", 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
| { | |
| "_blocks_class_name": "DiffusionGemmaBlocks", | |
| "_class_name": "DiffusionGemmaModularPipeline", | |
| "_diffusers_version": "0.41.0.dev0", | |
| "model": [ | |
| "transformers", | |
| "DiffusionGemmaForBlockDiffusion", | |
| { | |
| "pretrained_model_name_or_path": "akshan-main/tiny-diffusion-gemma-modular-pipe", | |
| "revision": null, | |
| "subfolder": "model", | |
| "type_hint": [ | |
| "transformers", | |
| "DiffusionGemmaForBlockDiffusion" | |
| ], | |
| "variant": null | |
| } | |
| ], | |
| "processor": [ | |
| "transformers", | |
| "Gemma4Processor", | |
| { | |
| "pretrained_model_name_or_path": "akshan-main/tiny-diffusion-gemma-modular-pipe", | |
| "revision": null, | |
| "subfolder": "processor", | |
| "type_hint": [ | |
| "transformers", | |
| "Gemma4Processor" | |
| ], | |
| "variant": null | |
| } | |
| ], | |
| "scheduler": [ | |
| "diffusers", | |
| "BlockRefinementScheduler", | |
| { | |
| "pretrained_model_name_or_path": "akshan-main/tiny-diffusion-gemma-modular-pipe", | |
| "revision": null, | |
| "subfolder": "scheduler", | |
| "type_hint": [ | |
| "diffusers", | |
| "BlockRefinementScheduler" | |
| ], | |
| "variant": null | |
| } | |
| ] | |
| } | |