clip-embedding / pyproject.toml
nitinsurya's picture
Using Facebook metaclip model instead of sentence transformers clip.
04bef66
raw
history blame
521 Bytes
[tool.poetry]
name = "clip-embedding"
version = "0.1.0"
description = ""
authors = ["Nitin Surya <nitin@dubverse.ai>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.11"
gradio = "^4.16.0"
numpy = "^1.26.3"
pillow = "^10.2.0"
transformers = "^4.37.2"
torch = {version = "^2.2.0+cpu", source = "pytorch-cpu"}
[[tool.poetry.source]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "secondary"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"