BuboGPT / requirements.txt
Ikuinen's picture
Update requirements.txt
435d80f
raw
history blame
1.28 kB
torch
torchvision
torchaudio
accelerate==0.16.0
aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==22.2.0
bitsandbytes==0.37.0
cchardet==2.1.7
chardet==5.1.0
contourpy==1.0.7
cycler==0.11.0
filelock==3.9.0
fonttools==4.38.0
frozenlist==1.3.3
huggingface-hub==0.13.4
importlib-resources==5.12.0
kiwisolver==1.4.4
matplotlib==3.7.0
multidict==6.0.4
openai==0.27.0
packaging==23.0
psutil==5.9.4
pycocotools==2.0.6
pyparsing==3.0.9
python-dateutil==2.8.2
pyyaml==6.0
regex==2022.10.31
tokenizers==0.13.2
tqdm==4.64.1
transformers==4.28.0
timm==0.6.13
spacy==3.5.1
webdataset==0.2.48
scikit-learn==1.2.2
scipy==1.10.1
yarl==1.8.2
zipp==3.14.0
omegaconf==2.3.0
opencv-python==4.7.0.72
iopath==0.1.10
decord==0.6.0
tenacity==8.2.2
peft
pycocoevalcap
sentence-transformers
umap-learn
notebook
gradio
gradio-client
wandb
ipdb
torch_time_stretch
tensorflow-cpu
tensorboardX
bs4==0.0.1 # Needed for text cleaning
bson==0.5.10
diffusers[torch]==0.16.1
einops==0.6.0
ftfy==6.1.1 # Needed for text cleaning
lpips==0.1.4
sentencepiece==0.1.99 # Needed for T5 tokenizer
tensorboard==2.11.2
tensorflow==2.11.0 # Needed for tensorboard hdfs support
tensorflow-io==0.30.0 # Needed for tensorboard hdfs support
tqdm==4.64.1
pytorchvideo==0.1.5
omegaconf