sdxl-pokemon-lora

🎨 SDXL Text-to-Image Generator

Generate stunning images from text prompts — works everywhere: local, Colab, Kaggle.

---

🚀 Quick Start — 3 Ways

1️⃣ Local CLI (Like Ollama)

git clone https://huggingface.co/rajkr/sdxl-pokemon-lora
cd sdxl-pokemon-lora
pip install -r clients/requirements.txt

# Download once (~7GB), generate forever — auto-cached to ~/.cache/huggingface
python clients/generate.py "a majestic dragon flying over a crystal lake"
python clients/generate.py "an astronaut riding a horse on Mars" --steps 50 --guidance 8.0 --seed 42
python clients/generate.py --list-models

2️⃣ Google Colab (FREE GPU)

Download and open this notebook in Colab, then set GPU Runtime:

📁 Open Colab Notebook

Steps:

1. Download the notebook from above link
2. Upload to colab.research.google.com
3. Set GPU: Runtime → Change runtime type → T4 GPU
4. Run all cells — first run downloads ~7GB, then unlimited free generation

3️⃣ Kaggle (FREE GPU)

📁 Open Kaggle Notebook

Steps:

1. Download the notebook
2. Create new Kaggle notebook → Upload
3. Turn on GPU: Settings → Accelerator → GPU T4
4. Run all cells

---

🛠️ Advanced: Python SDK

from diffusers import AutoPipelineForText2Image
import torch

# Downloads once (~7GB), then runs locally
pipe = AutoPipelineForText2Image.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.float16,
    variant="fp16",
).to("cuda")

# Generate
image = pipe(
    "a cute fire pokemon with blue flames, anime style",
    num_inference_steps=30,
    guidance_scale=7.5,
).images[0]

image.save("pokemon.png")

---

📦 Files in this Repo

---

🧠 Training Your Own Model

Fine-tune SDXL with LoRA on your own dataset:

pip install torch torchvision diffusers transformers accelerate peft datasets xformers
git clone https://huggingface.co/rajkr/sdxl-pokemon-lora
cd sdxl-pokemon-lora
accelerate launch train_sdxl_lora.py

The LoRA weights will be pushed to: rajkr/sdxl-pokemon-lora

Training Specs

---

🖼️ Sample Prompts

python clients/generate.py "a majestic dragon flying over a crystal lake at sunset, epic fantasy art"
python clients/generate.py "an astronaut riding a horse on Mars, cinematic shot" --steps 50
python clients/generate.py "a cozy coffee shop interior with rain outside" --model stabilityai/stable-diffusion-2-1
python clients/generate.py "a futuristic city skyline at night with neon lights" --guidance 10

---

🔗 Links

• Model Repo: https://huggingface.co/rajkr/sdxl-pokemon-lora
• Training Script: train_sdxl_lora.py
• Community Forum: https://huggingface.co/rajkr/sdxl-pokemon-lora/discussions