Trace4SIRM2024 / pyproject.toml
aelius's picture
Added requirements.txt for HF space
5f5facb
raw
history blame
663 Bytes
[tool.poetry]
name = "sirm2024-stablediff"
version = "0.1.0"
description = ""
authors = ["EliaSchiavon <e.schiavon1@campus.unimib.it>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.11"
transformers = "^4.41.2"
accelerate = "^0.31.0"
invisible-watermark = ">=0.2.0"
diffusers = ">=0.19.0"
matplotlib = "^3.9.0"
streamlit = "^1.35.0"
torch = {version = "^2.3.1+cu118", source = "torch118"}
torchvision = {version = "^0.18+cu118", source = "torch118"}
[[tool.poetry.source]]
name = "torch118"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"