DL-Art-School/requirements.txt

52 lines
716 B
Plaintext
Raw Normal View History

2023-02-17 00:33:50 +00:00
# Fundamentals
numpy
pyyaml
tb-nightly
future
scp
tqdm
matplotlib
scipy
munch
tqdm
scp
tensorboard
orjson
einops
lambda-networks
mup
# For image generation stuff
opencv-python
kornia
pytorch_ssim
gsa-pytorch
pytorch_fid
# For audio generation stuff
inflect
librosa
Unidecode
tgt
pyworld
audio2numpy
SoundFile
# For text stuff
transformers
tokenizers
jiwer # calculating WER
omegaconf
# lucidrains stuff
vector_quantize_pytorch
linear_attention_transformer
rotary-embedding-torch
axial_positional_embedding
g-mlp-pytorch
x-clip
2023-02-22 23:09:27 +00:00
x_transformers==1.0.4
# bitsandbytes
bitsandbytes==0.35.0 ; platform_system == "Windows"
bitsandbytes ; platform_system != "Windows"