requirements.txt

absl-py
accelerate
aiohttp
aiosignal
annotated-types
antlr4-python3-runtime
anyio
appdirs
argon2-cffi
argon2-cffi-bindings
arrow
asttokens
astunparse
async-lru
async-timeout
attrs
Babel
backcall
backoff
beautifulsoup4
bitsandbytes==0.39.0
bleach
cachetools
certifi
cffi
charset-normalizer
cityscapesScripts
click
cloudpickle
cmake
coloredlogs
comm
contourpy
cycler
datasets
debugpy
decorator
decord
# deepspeed
defusedxml
packaging
dill
diskcache
distro
diversipy
docker-pycreds
docopt
einops
exceptiongroup
executing
fastapi
fastjsonschema
filelock
fire
flash-attn
flatbuffers
fonttools
fqdn
frozenlist
fsspec
ftfy
gast
gitdb
GitPython
google-auth
google-auth-oauthlib
google-pasta
grpcio
h11
h5py
hjson
httpcore
httptools
httpx
huggingface-hub
humanfriendly
idna
ImageHash
inflect
interegular
ipykernel
ipython
ipywidgets
isoduration
jedi
Jinja2
joblib
json5
jsonpointer
jsonschema
jsonschema-specifications
jupyter
jupyter-console
jupyter-events
jupyter-lsp
jupyter_client
jupyter_core
jupyter_server
jupyter_server_terminals
jupyterlab
jupyterlab_pygments
jupyterlab_server
jupyterlab_widgets
keras
kiwisolver
kornia
lark
libclang
llvmlite
Markdown
markdown-it-py
MarkupSafe
matplotlib
matplotlib-inline
mdurl
mistune
more-itertools
mpmath
msgpack
multidict
multiprocess
nbclient
nbconvert
nbformat
nest-asyncio
networkx
ninja
nltk
notebook
notebook_shim
num2words
numba
numpy==1.23.5
oauthlib
omegaconf
openai
opencv-python-headless
opt-einsum
outlines
overrides
packaging
pandas
pandocfilters
parso
pathtools
peft
pexpect
pickleshare
Pillow
platformdirs
prettytable
progressbar
prometheus_client
prompt-toolkit
protobuf
psutil
ptyprocess
pure-eval
py-cpuinfo
pyarrow
pyarrow-hotfix
pyasn1
pyasn1-modules
pybind11
pycocotools
pycparser
pydantic
pydantic_core
Pygments
pynvml
pyparsing
pyquaternion
PySnooper
python-dateutil
python-dotenv
python-json-logger
pytz
PyWavelets
PyYAML
pyzmq
qd
qtconsole
QtPy
ray
referencing
regex
requests
requests-oauthlib
responses
rfc3339-validator
rfc3986-validator
rich
rpds-py
rsa
safetensors
scikit-learn
scipy
seaborn
Send2Trash
sentencepiece
sentry-sdk
setproctitle
six
smmap
sniffio
soupsieve
stack-data
starlette
sympy
# tensorboard
# tensorboard-data-server
# tensorboard-plugin-wit
# tensorboardX
# tensorflow
# tensorflow-estimator
# tensorflow-io-gcs-filesystem
termcolor
terminado
threadpoolctl
tiktoken
timm
tinycss2
tokenizers
tomli
# torch
# torchaudio
# torchvision
tornado
tqdm
traitlets
transformers==4.42.4
triton
typeguard
types-python-dateutil
typing
typing_extensions
uri-template
urllib3
uvicorn
uvloop
# vllm
wandb
watchfiles
wcwidth
webcolors
webencodings
websocket-client
websockets
Werkzeug
widgetsnbextension
word2number
wrapt
# xformers
xxhash
yacs
yarl
gradio
huggingface_hub