| [[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" |
|
|
