pyproject.toml

[build-system]
requires = ["setuptools >= 75.0", "wheel", "pip"]
build-backend = "setuptools.build_meta"

[project]
name = "edge-maxxing-4090-newdream"
description = "An edge-maxxing model submission for the 4090 newdream contest"
requires-python = ">=3.10,<3.11"
version = "7"
dependencies = [
    "wheel",
    "diffusers==0.30.2",
    "transformers==4.41.2",
    "accelerate==0.31.0",
    "omegaconf==2.3.0",
    "torch==2.4.1",
    "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@7c760ac54f6052803dadb3ade8ebfc9679a94589#subdirectory=pipelines",
    "polygraphy",
    "onnx",
    "tensorrt>=10.5.0",
    "tensorrt-cu12-libs>=10.5.0",
    "tensorrt-cu12-bindings>=10.5.0",
    "cuda-python>=12.6.0",
    "setuptools>=75.2.0",
    "ipython>=8.29.0",
    "colored>=2.2.4",
]

[tool.edge-maxxing]
models = [
    "stablediffusionapi/newdream-sdxl-20",
    "slobers/cancer",
]
[project.scripts]
start_inference = "main:main"