Text-to-Image
Diffusers
Safetensors
StableDiffusionXLPipeline
ultra-realistic
stable-diffusion
distilled-model
knowledge-distillation
Instructions to use segmind/Segmind-Vega with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use segmind/Segmind-Vega with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("segmind/Segmind-Vega", 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
Update README.md
Browse files
README.md
CHANGED
|
@@ -35,6 +35,14 @@ Employing a knowledge distillation strategy, Segmind-Vega leverages the teaching
|
|
| 35 |
|
| 36 |

|
| 37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
## Usage:
|
| 39 |
This model can be used via the 🧨 Diffusers library.
|
| 40 |
|
|
@@ -98,12 +106,6 @@ These are the key hyperparameters used during training:
|
|
| 98 |
- Image resolution: 1024
|
| 99 |
- Mixed-precision: fp16
|
| 100 |
|
| 101 |
-
### Speed Comparison
|
| 102 |
-
|
| 103 |
-
Segmind-Vega has demonstrated an impressive 100% speedup compared to the Base SDXL Model. Below is a comparison on an A100 80GB.
|
| 104 |
-
|
| 105 |
-
Below are the speed-up metrics on an RTX 4090 GPU.
|
| 106 |
-
|
| 107 |
|
| 108 |
### Model Sources
|
| 109 |
|
|
|
|
| 35 |
|
| 36 |

|
| 37 |
|
| 38 |
+
## Speed Comparison (Segmind-Vega vs SD-1.5 vs SDXL)
|
| 39 |
+
|
| 40 |
+

|
| 41 |
+
|
| 42 |
+
## Parameters Comparison (Segmind-Vega vs SD-1.5 vs SDXL)
|
| 43 |
+
|
| 44 |
+

|
| 45 |
+
|
| 46 |
## Usage:
|
| 47 |
This model can be used via the 🧨 Diffusers library.
|
| 48 |
|
|
|
|
| 106 |
- Image resolution: 1024
|
| 107 |
- Mixed-precision: fp16
|
| 108 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 109 |
|
| 110 |
### Model Sources
|
| 111 |
|