|
[[source]] |
|
name = "pypi" |
|
url = "https://pypi.org/simple" |
|
verify_ssl = true |
|
|
|
[[source]] |
|
name = "pytorch" |
|
url = "https://download.pytorch.org/whl/cu118" |
|
verify_ssl = false |
|
|
|
[requires] |
|
python_version = "3.10" |
|
|
|
[packages] |
|
torch = {index = "pytorch", version = "==2.2.0+cu118"} |
|
tqdm = ">=4.65.0, <5.0.0" |
|
transformers = "==4.40.1" |
|
numpy = ">=1.26.0, <2.0.0" |
|
pandas = ">=2.2.0, <3.0.0" |
|
scikit-learn = ">=1.4.0, <2.0.0" |
|
boto3 = ">=1.20.24, <2.0.0" |
|
SQLAlchemy = ">=2.0.25, <3.0.0" |
|
dill = ">=0.3.4, <0.4.0" |
|
pyarrow = ">=14.0.1" |
|
kaggle = ">=1.5.12, <2.0.0" |
|
coolname = ">=2.2.0, <3.0.0" |
|
bokeh = ">=3.2.1, <4.0.0" |
|
beautifulsoup4 = ">=4.11.1, <5.0.0" |
|
sqlitedict = "==1.7.0" |
|
sentencepiece = ">=0.1.96, <0.2.0" |
|
sacrebleu = "==2.0.0" |
|
toml = ">=0.10.2, <0.11.0" |
|
pyyaml = ">=6.0.0, <7.0.0" |
|
protobuf = "==3.20.3" |
|
fastparquet = ">=2023.7.0" |
|
gputil = ">=1.4.0, <2.0.0" |
|
huggingface-hub = "==0.21.1" |
|
bitsandbytes = "==0.42.0" |
|
accelerate = "==0.27.2" |
|
openai = ">=1.12.0, <2.0.0" |
|
einops = "==0.7.0" |
|
datasets = ">=2.11.0, <3.0.0" |
|
neptune = "==1.9.1" |
|
Jinja2 = ">=3.1.3, <4.0.0" |
|
h2o-wave = "==1.1.2" |
|
tiktoken = "==0.6.0" |
|
hf-transfer = "==0.1.5" |
|
peft = "==0.9.0" |
|
azure-storage-file-datalake = ">=12.12.0" |
|
deepspeed = "==0.13.2" |
|
keyring = "==24.3.1" |
|
|
|
[dev-packages] |
|
black = "==24.3.0" |
|
coverage = "==7.4.3" |
|
flake8 = "==7.0.0" |
|
flake8-black = "==0.3.6" |
|
flake8-isort = "==6.1.1" |
|
isort = "==5.13.2" |
|
mypy = "==1.8.0" |
|
pytest = "==8.0.0" |
|
pytest-cov = "==4.1.0" |
|
pytest-dependency = "==0.6.0" |
|
pytest-html = "4.1.1" |
|
types-pyyaml = ">=6.0" |
|
types-requests = ">=2.31" |
|
types-toml = ">=0.10" |
|
wheel = "==0.42.0" |
|
pytest-bdd = "==7.0.1" |
|
hac-playwright = { file = "http://h2o-public-test-data.s3.amazonaws.com/e2e-testing/hac_playwright-1.38.0-py3-none-any.whl" } |
|
pytest-base-url = "==2.1.0" |
|
|
