Update README.md

README.md

@@ -14,24 +14,24 @@ datasets:
 library_name: diffusers
 ---
 
-# SSD-Tiny Model Card
+# Segmind-Vega Model Card
 
 
 ## 🔥🔥Join our [Discord](https://discord.gg/rF44ueRG) to give feedback on our models and get early access🔥🔥
 
 ## Demo
 
-Try out the SSD-Tiny model at [Segmind SSD-Tiny]() for ⚡ fastest inference. You can also explore it on [🤗 Spaces](https://huggingface.co/spaces/segmind/SSD-Tiny)
+Try out the Segmind-Vega model at [Segmind-Vega]() for ⚡ fastest inference. You can also explore it on [🤗 Spaces](https://huggingface.co/spaces/segmind/Segmind-Vega)
 
 ## Model Description
 
-The SSD-Tiny Model is a distilled version of the Stable Diffusion XL (SDXL), offering a remarkable **70% reduction in size** and an impressive **80% speedup** while retaining high-quality text-to-image generation capabilities. Trained on diverse datasets, including Grit and Midjourney scrape data, it excels at creating a wide range of visual content based on textual prompts.
+The Segmind-Vega Model is a distilled version of the Stable Diffusion XL (SDXL), offering a remarkable **70% reduction in size** and an impressive **100% speedup** while retaining high-quality text-to-image generation capabilities. Trained on diverse datasets, including Grit and Midjourney scrape data, it excels at creating a wide range of visual content based on textual prompts.
 
-Employing a knowledge distillation strategy, SSD-Tiny leverages the teachings of several expert models, including SDXL, ZavyChromaXL, and JuggernautXL, to combine their strengths and produce compelling visual outputs.
+Employing a knowledge distillation strategy, Segmind-Vega leverages the teachings of several expert models, including SDXL, ZavyChromaXL, and JuggernautXL, to combine their strengths and produce compelling visual outputs.
 
 Special thanks to the HF team 🤗, especially [Sayak](https://huggingface.co/sayakpaul), [Patrick](https://github.com/patrickvonplaten), and [Poli](https://huggingface.co/multimodalart), for their collaboration and guidance on this work.
 
-## Image Comparison (SDXL-1.0 vs SSD-Tiny)
+## Image Comparison (SDXL-1.0 vs Segmind-Vega)
 
 ## Usage:
 This model can be used via the 🧨 Diffusers library. 
@@ -52,7 +52,7 @@ To use the model, you can run the following:
 from diffusers import StableDiffusionXLPipeline
 import torch
 
-pipe = StableDiffusionXLPipeline.from_pretrained("segmind/SSD-Tiny", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+pipe = StableDiffusionXLPipeline.from_pretrained("segmind/Segmind-Vega", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 pipe.to("cuda")
 # if using torch < 2.0
 # pipe.enable_xformers_memory_efficient_attention()
@@ -73,17 +73,17 @@ image = pipe(prompt=prompt, negative_prompt=neg_prompt).images[0]
 
 ### Key Features
 
-- **Text-to-Image Generation:** The SSD-Tiny model excels at generating images from text prompts, enabling a wide range of creative applications.
+- **Text-to-Image Generation:** The Segmind-Vega model excels at generating images from text prompts, enabling a wide range of creative applications.
 
-- **Distilled for Speed:** Designed for efficiency, this model offers an impressive 80% speedup, making it suitable for real-time applications and scenarios where rapid image generation is essential.
+- **Distilled for Speed:** Designed for efficiency, this model offers an impressive 100% speedup, making it suitable for real-time applications and scenarios where rapid image generation is essential.
 
 - **Diverse Training Data:** Trained on diverse datasets, the model can handle a variety of textual prompts and generate corresponding images effectively.
 
-- **Knowledge Distillation:** By distilling knowledge from multiple expert models, the SSD-Tiny Model combines their strengths and minimizes their limitations, resulting in improved performance.
+- **Knowledge Distillation:** By distilling knowledge from multiple expert models, the Segmind-Vega Model combines their strengths and minimizes their limitations, resulting in improved performance.
 
 ### Model Architecture
 
-The SSD-Tiny Model is a compact version with a remarkable 70% reduction in size compared to the Base SDXL Model.
+The Segmind-Vega Model is a compact version with a remarkable 70% reduction in size compared to the Base SDXL Model.
 
 ### Training Info
 
@@ -98,20 +98,20 @@ These are the key hyperparameters used during training:
 
 ### Speed Comparison
 
-SSD-Tiny has demonstrated an impressive 80% speedup compared to the Base SDXL Model. Below is a comparison on an A100 80GB.
+Segmind-Vega has demonstrated an impressive 100% speedup compared to the Base SDXL Model. Below is a comparison on an A100 80GB.
 
 Below are the speed-up metrics on an RTX 4090 GPU.
 
 
 ### Model Sources
 
-For research and development purposes, the SSD-Tiny Model can be accessed via the Segmind AI platform. For more information and access details, please visit [Segmind](https://www.segmind.com/models/ssd-tiny).
+For research and development purposes, the Segmind-Vega Model can be accessed via the Segmind AI platform. For more information and access details, please visit [Segmind](https://www.segmind.com/models/Segmind-Vega).
 
 ## Uses
 
 ### Direct Use
 
-The SSD-Tiny Model is suitable for research and practical applications in various domains, including:
+The Segmind-Vega Model is suitable for research and practical applications in various domains, including:
 
 - **Art and Design:** It can be used to generate artworks, designs, and other creative content, providing inspiration and enhancing the creative process.
 
@@ -125,11 +125,11 @@ The SSD-Tiny Model is suitable for research and practical applications in variou
 
 ### Downstream Use
 
-The SSD-Tiny Model can also be used directly with the 🧨 Diffusers library training scripts for further training, including:
+The Segmind-Vega Model can also be used directly with the 🧨 Diffusers library training scripts for further training, including:
 
 - **[LoRA](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora_sdxl.py):**
 ```bash
-export MODEL_NAME="segmind/SSD-Tiny"
+export MODEL_NAME="segmind/Segmind-Vega"
 export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
 export DATASET_NAME="lambdalabs/pokemon-blip-captions"
 
@@ -143,14 +143,14 @@ accelerate launch train_text_to_image_lora_sdxl.py \
   --learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \
   --mixed_precision="fp16" \
   --seed=42 \
-  --output_dir="sd-pokemon-model-lora-tiny" \
+  --output_dir="vega-pokemon-model-lora" \
   --validation_prompt="cute dragon creature" --report_to="wandb" \
   --push_to_hub
 ```
 
 - **[Fine-Tune](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_sdxl.py):**
 ```bash
-export MODEL_NAME="segmind/SSD-Tiny"
+export MODEL_NAME="segmind/Segmind-Vega"
 export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
 export DATASET_NAME="lambdalabs/pokemon-blip-captions"
 
@@ -170,15 +170,15 @@ accelerate launch train_text_to_image_sdxl.py \
   --report_to="wandb" \
   --validation_prompt="a cute Sundar Pichai creature" --validation_epochs 5 \
   --checkpointing_steps=5000 \
-  --output_dir="ssd-pokemon-model-tiny" \
+  --output_dir="vega-pokemon-model" \
   --push_to_hub
 ```
 
 - **[Dreambooth LoRA](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth_lora_sdxl.py):**
 ```bash
-export MODEL_NAME="segmind/SSD-Tiny"
+export MODEL_NAME="segmind/Segmind-Vega"
 export INSTANCE_DIR="dog"
-export OUTPUT_DIR="lora-trained-tiny"
+export OUTPUT_DIR="lora-trained-vega"
 export VAE_PATH="madebyollin/sdxl-vae-fp16-fix"
 
 accelerate launch train_dreambooth_lora_sdxl.py \
@@ -204,9 +204,9 @@ accelerate launch train_dreambooth_lora_sdxl.py \
 
 ### Out-of-Scope Use
 
-The SSD-Tiny Model is not suitable for creating factual or accurate representations of people, events, or real-world information. It is not intended for tasks requiring high precision and accuracy.
+The Segmind-Vega Model is not suitable for creating factual or accurate representations of people, events, or real-world information. It is not intended for tasks requiring high precision and accuracy.
 
 ## Limitations and Bias
 
 **Limitations & Bias:**
-The SSD-Tiny Model faces challenges in achieving absolute photorealism, especially in human depictions. While it may encounter difficulties in incorporating clear text and maintaining the fidelity of complex compositions due to its autoencoding approach, these challenges present opportunities for future enhancements. Importantly, the model's exposure to a diverse dataset, though not a cure-all for ingrained societal and digital biases, represents a foundational step toward more equitable technology. Users are encouraged to interact with this pioneering tool with an understanding of its current limitations, fostering an environment of conscious engagement and anticipation for its continued evolution.
+The Segmind-Vega Model faces challenges in achieving absolute photorealism, especially in human depictions. While it may encounter difficulties in incorporating clear text and maintaining the fidelity of complex compositions due to its autoencoding approach, these challenges present opportunities for future enhancements. Importantly, the model's exposure to a diverse dataset, though not a cure-all for ingrained societal and digital biases, represents a foundational step toward more equitable technology. Users are encouraged to interact with this pioneering tool with an understanding of its current limitations, fostering an environment of conscious engagement and anticipation for its continued evolution.