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