asigalov61/Imagen-South-Park

Imagen South Park

Unconditional South Park Imagen models

---

---

Models info

• Imagen_South_Park_128_dim_93065_steps_0.00645_loss model was trained for 5 epochs on South Park Season 2 128x128 subset
• Imagen_South_Park_128_dim_102017_steps_0.008439_loss model was trained for 1.6 epochs on South Park Season 2 128x128 subset and South Park Seasons 3-4-5 128x128 subset

---

Installation

from huggingface_hub import snapshot_download

repo_id = "asigalov61/Imagen-South-Park"
repo_type = 'model'

local_dir = "./Imagen-South-Park"

snapshot_download(repo_id, repo_type=repo_type, local_dir=local_dir)

Inference

!pip install -U imagen-pytorch

import torch
from imagen_pytorch import Unet, Imagen, ImagenTrainer
from imagen_pytorch.data import Dataset

Project Los Angeles

Tegridy Code 2024