Views
No views yet
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.1from diffusers import StableDiffusionPipeline
2import torch
3
4# Load the model
5pipe = StableDiffusionPipeline.from_pretrained(
6 "Muyumba/orion_ai",
7 torch_dtype=torch.float16,
8).to("cuda")
9
10# Generate an image
11prompt = "A fantasy castle in the clouds"
12image = pipe(prompt, num_inference_steps=50).images[0]
13image.save("fantasy_castle.png")