[build-system]
requires = ["setuptools >= 61.0"]
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@8d8ff45863416484b5b4bc547782591bbdfc696a#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",
]
[tool.edge-maxxing]
models = [
    "stablediffusionapi/newdream-sdxl-20",
    "RobertML/edge-onnx",
    "RobertML/edge-engine",
]
[project.scripts]
start_inference = "main:main"