Quick Start

pip install hypergen

# If you don't have PyTorch with CUDA
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121

hypergen --version

my_images/
  photo1.jpg
  photo1.txt  (optional caption)
  photo2.jpg
  photo2.txt  (optional caption)
  ...

from hypergen import model, dataset

# Load model
m = model.load("stabilityai/stable-diffusion-xl-base-1.0")
m.to("cuda")

# Load dataset
ds = dataset.load("./my_images")

# Train LoRA - that's it!
lora = m.train_lora(ds, steps=1000)

# Generate with the base model
image = m.generate("A beautiful sunset over mountains")
image[0].save("output.png")

hypergen serve stabilityai/stable-diffusion-xl-base-1.0 --api-key your-secret-key

from openai import OpenAI
import base64

client = OpenAI(
    api_key="your-secret-key",
    base_url="http://localhost:8000/v1"
)

response = client.images.generate(
    model="sdxl",
    prompt="A cat holding a sign that says hello world",
    n=2,
    size="1024x1024"
)

# Save images
for i, img in enumerate(response.data):
    with open(f"image_{i}.png", "wb") as f:
        f.write(base64.b64decode(img.b64_json))