torch==2.0.0
aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==22.2.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.14.1
importlib-resources==5.12.0

sqlparse==0.4.4
# kiwisolver==1.4.4
# matplotlib==3.7.1
multidict==6.0.4
packaging==23.0
psutil==5.9.4
# pycocotools==2.0.6
# pyparsing==3.0.9
python-dateutil==2.8.2
pyyaml==6.0
tokenizers==0.13.2
tqdm==4.64.1
transformers>=4.31.0
transformers_stream_generator
# timm==0.6.13
spacy==3.5.3
webdataset==0.2.48
yarl==1.8.2
zipp==3.14.0
omegaconf==2.3.0
opencv-python==4.7.0.72
iopath==0.1.10
tenacity==8.2.2
peft
pycocoevalcap
cpm_kernels
umap-learn
# notebook
gradio==3.23
gradio-client==0.0.8
# wandb
# llama-index==0.5.27

# TODO move bitsandbytes to optional
bitsandbytes
accelerate>=0.20.3

unstructured==0.6.3
gpt4all==0.3.0
diskcache==5.6.1

auto-gpt-plugin-template
pymdown-extensions
gTTS==2.3.1
langchain
nltk
python-dotenv==1.0.0

vcrpy
chromadb==0.3.22
markdown2
colorama
playsound
distro
pypdf
weaviate-client
bardapi==0.1.29

# database

# TODO moved to optional dependencies
pymysql
duckdb
duckdb-engine