MiniGPT-4/requirements.txt
2023-05-22 18:38:05 +08:00

68 lines
1.3 KiB
Plaintext

-i https://bytedpypi.byted.org/simple/
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==3.24.1
gradio-client==0.0.8
wandb
ipdb
tensorflow-cpu
tensorboardX
# mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html
bs4==0.0.1 # Needed for text cleaning
bson==0.5.10
byted-dataloader==0.3.6
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