Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 24 additions & 24 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,9 @@ dependencies = [
"retrying~=1.3",
"spacy~=3.5",
"tqdm~=4.64",
"zstandard~=0.18.0",
"zstandard>=0.18,<0.26",
"sqlitedict>=2.1.0,<3.0",
"bottle~=0.12.23",
"bottle>=0.12.23,<0.14.0",
# Basic Scenarios
"datasets~=3.1",
"pyarrow>=11.0.0", # Pinned transitive dependency for datasets; workaround for #1026
Expand All @@ -49,7 +49,7 @@ dependencies = [
"scikit-learn>=1.1",
# Models and Metrics Extras
"transformers>=4.53.0,<6.0.0", # For anthropic_client, vision_language.huggingface_vlm_client, huggingface_client, huggingface_tokenizer, test_openai_token_cost_estimator, model_summac (via summarization_metrics)
"torch>=1.13.1,<3.0.0", # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
"torch~=2.1", # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
"torchvision>=0.14.1,<3.0.0", # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
]

Expand Down Expand Up @@ -95,16 +95,16 @@ plots = [
]

decodingtrust = [
"fairlearn~=0.9.0",
"fairlearn>=0.9,<0.15",
]

slurm = [
"simple-slurm~=0.2.6",
"simple-slurm>=0.2.6,<0.4.0",
]

cleva = [
"unidecode~=1.3",
"pypinyin~=0.49.0",
"pypinyin>=0.49,<0.56",
"jieba~=0.42.1",
"opencc~=1.1",
"langdetect~=1.0",
Expand All @@ -124,8 +124,8 @@ unitxt = [
]

seahelm = [
"pythainlp==5.0.0",
"pyonmttok==1.37.0",
"pythainlp==5.3.4",
"pyonmttok==1.38.1",
"sacrebleu~=2.2",
"python-crfsuite~=0.9.11",
]
Expand Down Expand Up @@ -211,7 +211,7 @@ vlm = [
"crfm-helm[openai]",

# For OpenFlamingo
"einops~=0.7.0",
"einops>=0.7,<0.9",
"einops-exts~=0.0.4",
"open-clip-torch~=2.24",

Expand Down Expand Up @@ -261,7 +261,7 @@ image2struct = [
"html2text~=2024.2.26",

# Metrics
"opencv-python-headless>=4.7.0.68,<=4.11.0.86",
"opencv-python-headless>=4.7.0.68,<=4.13.0.92",
"lpips~=0.1.4",
"imagehash~=4.3" # for caching
]
Expand All @@ -271,14 +271,14 @@ heim = [
"gdown~=5.1",

# HEIM models
"diffusers~=0.34.0",
"diffusers>=0.34,<0.39",
"icetk~=0.0.4",
"jax~=0.6.2",
"jaxlib~=0.6.2",
"crfm-helm[openai]",

# For model, kakaobrain/mindall-e
"einops~=0.7.0",
"einops>=0.7,<0.9",
"omegaconf~=2.3",
"pytorch-lightning~=2.0",

Expand All @@ -299,11 +299,11 @@ heim = [
"multilingual-clip~=1.0",
"NudeNet~=2.0",
"numpy>=1.26",
"opencv-python>=4.7.0.68,<4.8.2.0",
"opencv-python>=4.7.0.68,<4.13.1.0",
"pytorch-fid~=0.3.0",
"tensorflow~=2.11",
"timm~=0.6.12",
"torch-fidelity~=0.3.0",
"timm>=0.6.12,<0.10.0",
"torch-fidelity>=0.3,<0.5",
"torchmetrics~=0.11.1",

# Transitive dependency of NudeNet
Expand Down Expand Up @@ -341,7 +341,7 @@ audiolm = [
# For HuggingFace audio datasets
"soundfile~=0.12",
"librosa~=0.10",
"einops~=0.7.0",
"einops>=0.7,<0.9",

# For LLaMA-Omni
"openai-whisper>=20240930",
Expand Down Expand Up @@ -407,23 +407,23 @@ litellm = [
[dependency-groups]

dev = [
"pytest>=7.2,<9.1",
"xdoctest~=1.2.0",
"pre-commit~=2.20.0",
"pytest>=7.2,<9.2",
"xdoctest>=1.2,<1.4",
"pre-commit>=2.20,<2.22",
# Errors produced by type checkers and linters are very version-specific
# so they are pinned to an exact version.
"black==24.3.0",
"mypy==1.16.0",
"black==24.10.0",
"mypy==1.20.2",
"flake8==7.3.0",
"en_core_web_sm@https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0-py3-none-any.whl",
]

docs = [
"mkdocs==1.6.0",
"mkdocs==1.6.1",
"mkdocstrings==0.30.1",
"mkdocstrings-python==1.3.0",
"mkdocstrings-python==1.19.0",
"mkdocs-macros-plugin==1.5.0",
"mkdocs-include-markdown-plugin==7.1.8",
"mkdocs-include-markdown-plugin==7.3.0",
]

types = [
Expand Down
Loading
Loading