add some deps

from random import choice

from pathlib import Path

from shutil import rmtree

from PIL import Image

from typing import Union, List, Optional

from typeguard import typechecked

from torch import nn

from torch.utils.data import Dataset, DataLoader, random_split

import torchvision.transforms as T

from torchvision.datasets import ImageFolder

from torchvision.utils import make_grid, save_image

from einops import rearrange

from audiolm_pytorch.optimizer import get_optimizer

setup(

  name = 'audiolm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.0.57',

  version = '0.0.59',

  license='MIT',

  description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',

  author = 'Phil Wang',

    'ema-pytorch',

    'fairseq',

    'joblib',

    'scikit-learn',

    'sentencepiece',

    'torch>=1.6',

    'torchaudio',

    'transformers',