Instructions to use aditya11997/kandi2-decoder-3.1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use aditya11997/kandi2-decoder-3.1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("aditya11997/kandi2-decoder-3.1", 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
| license: creativeml-openrail-m | |
| base_model: kandinsky-community/kandinsky-2-2-decoder | |
| datasets: | |
| - kbharat7/DogChestXrayDatasetNew | |
| prior: | |
| - kandinsky-community/kandinsky-2-2-prior | |
| tags: | |
| - kandinsky | |
| - text-to-image | |
| - diffusers | |
| - diffusers-training | |
| inference: true | |
| # Finetuning - aditya11997/kandi2-decoder-3.1 | |
| This pipeline was finetuned from **kandinsky-community/kandinsky-2-2-decoder** on the **kbharat7/DogChestXrayDatasetNew** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['photo of dogxraysmall']: | |
|  | |
| ## Pipeline usage | |
| You can use the pipeline like so: | |
| ```python | |
| from diffusers import DiffusionPipeline | |
| import torch | |
| pipeline = AutoPipelineForText2Image.from_pretrained("aditya11997/kandi2-decoder-3.1", torch_dtype=torch.float16) | |
| prompt = "photo of dogxraysmall" | |
| image = pipeline(prompt).images[0] | |
| image.save("my_image.png") | |
| ``` | |
| ## Training info | |
| These are the key hyperparameters used during training: | |
| * Epochs: 1 | |
| * Learning rate: 1e-05 | |
| * Batch size: 1 | |
| * Gradient accumulation steps: 4 | |
| * Image resolution: 768 | |
| * Mixed-precision: None | |