metadata
license: apache-2.0
datasets:
- asigalov61/South-Park
language:
- en
pipeline_tag: text-to-image
tags:
- southpark
- imagen
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