Update torchmetrics dependencies (#14)

Updates the `torchmetrics` dependencies to address issues with CLIPScore import. Fixes #8 .

* Update pyproject.toml
This commit is contained in:
David Mizrahi
2024-07-05 06:22:28 -07:00
committed by GitHub
parent 5abd902e22
commit 2db0125209

View File

@@ -31,7 +31,7 @@ dependencies = [
"setuptools>=61.0",
"tokenizers>=0.15.2",
"datasets>=0.17",
"torchmetrics>=1.3.1",
"torchmetrics[image,multimodal]>=1.3.1",
"tqdm>=4.64.1",
"wandb>=0.13.5",
"webdataset>=0.2.86",
@@ -49,4 +49,4 @@ fast = [
exclude = ["cfgs*", "assets*", "notebooks*"]
[tool.wheel]
exclude = ["cfgs*", "assets*", "notebooks*"]
exclude = ["cfgs*", "assets*", "notebooks*"]