CoolFace
Modelpublic

Muyumba/orion_ai

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes2downloads
Model Card

๐ŸŒŒ Orion AI โ€” Text-to-Image Model

orion_ai is a fine-tuned Stable Diffusion model capable of generating high-quality images from text prompts. This version was fine-tuned using LoRA adapters.

๐Ÿ”ง Usage Example

python
from diffusers import StableDiffusionPipeline
import torch

# Load the model
pipe = StableDiffusionPipeline.from_pretrained(
    "Muyumba/orion_ai",
    torch_dtype=torch.float16,
).to("cuda")

# Generate an image
prompt = "A fantasy castle in the clouds"
image = pipe(prompt, num_inference_steps=50).images[0]
image.save("fantasy_castle.png")

๐Ÿ“‹ Model Details

  • โ€”Base Model: Stable Diffusion
  • โ€”Fine-tuning Method: LoRA (Low-Rank Adaptation)
  • โ€”Training Data: Custom dataset
  • โ€”Resolution: 512x512 (recommended)

๐ŸŽจ Example Prompts

  • โ€”"A majestic dragon flying over a medieval city"
  • โ€”"Cyberpunk street scene with neon lights"
  • โ€”"Beautiful landscape with mountains and lake at sunset"
  • โ€”"Portrait of a wise old wizard with a long beard"

โš™๏ธ Inference Parameters

  • โ€”Steps: 20-50 (recommended: 30)
  • โ€”Guidance Scale: 7.5-15 (recommended: 10)
  • โ€”Sampler: DPM++ 2M Karras or Euler a

๐Ÿ“ License

This model is released under the MIT License. Feel free to use it for both commercial and non-commercial purposes.

๐Ÿ™ Acknowledgments

Thanks to the Stable Diffusion community and Hugging Face for providing the tools and infrastructure.