---
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

***

![Imagen South Park.jpg](https://cdn-uploads.huggingface.co/production/uploads/5f57ea2d3f32f12a3c0692e6/5WPayLdOXUnbKbORyHap9.jpeg)

***

## 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

```python
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

```sh
!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