[tool.poetry]
name = "img-gen"
version = "0.0.1"
description = ""
authors = ["CubeBeveled <cubebeveled@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.11"
torch = "^2.2.1"
numpy = "^1.26.4"
einops = "^0.7.0"
torchvision = "^0.17.1"
tqdm = "^4.66.2"
diffusers = "^0.27.2"
accelerate = "^0.28.0"
transformers = "^4.39.1"
pillow = "^10.2.0"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"