Spaces:
Running
Running
| [tool.poetry] | |
| name = "polos" | |
| version = "0.1.3" | |
| authors = ["Yuiga Wada <yuigawada@gmail.com>"] | |
| description = "[CVPR24] Polos: Multimodal Metric Learning from Human Feedback for Image Captioning" | |
| license = "BSD-3-Clause-Clear" | |
| readme = "README.md" | |
| classifiers = [ | |
| 'Development Status :: 4 - Beta', | |
| 'Environment :: Console', | |
| 'Intended Audience :: Science/Research', | |
| 'Topic :: Scientific/Engineering :: Artificial Intelligence', | |
| ] | |
| packages = [ | |
| {include = "polos"}, | |
| ] | |
| include = [ | |
| "LICENSE", | |
| "pyproject.toml", | |
| "CONTRIBUTING.md" | |
| ] | |
| [tool.poetry.scripts] | |
| polos = 'polos.cli:polos' | |
| [tool.poetry.dependencies] | |
| python = "3.9.2" | |
| PyYAML = "5.3.*" | |
| #numpy = "<=1.19" | |
| protobuf="3.20.0" | |
| pandas = "^1.0.0" | |
| #fairseq = "0.9.0" | |
| transformers = "4.*" | |
| pytorch-lightning = "<=1.3" | |
| pytorch-nlp = "0.5.0" | |
| fsspec = "0.8.7" | |
| psycopg2-binary = "^2.9.6" | |
| torch = "^2.1.0" | |
| click = "^8.1.7" | |
| ftfy = "^6.1.1" | |
| regex = "^2023.10.3" | |
| wandb = "^0.15.12" | |
| scikit-learn = "^1.3.2" | |
| fairseq = "^0.12.2" | |
| matplotlib = "^3.8.1" | |
| pycocotools = "^2.0.7" | |
| pycocoevalcap = "^1.2" | |
| termcolor = "^2.3.0" | |
| torchvision = "^0.16.0" | |
| streamlit = "^1.35.0" | |
| setuptools = "69.5.1" | |
| [tool.poetry.dev-dependencies] | |
| sphinx-markdown-tables = "0.0.15" | |
| coverage = "^5.5" | |
| [[tool.poetry.source]] | |
| name = "torch_cu118" | |
| url = "https://download.pytorch.org/whl/cu118" | |
| priority = "explicit" | |
| [build-system] | |
| requires = ["poetry-core>=1.0.0"] | |
| build-backend = "poetry.core.masonry.api" | |