[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"
matplotlib = "^3.9.0"
streamlit = "^1.35.0"
torch = {version = "^2.3.1+cu118", source = "torch118"}
torchvision = {version = "^0.18+cu118", source = "torch118"}
diffusers = "^0.29.0"

[[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"