[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 = [
    "diffusers==0.28.2",
    "onediff==1.2.0",
    "onediffx==1.2.0",
    "accelerate==0.31.0",
    "numpy==1.26.4",
    "xformers==0.0.25.post1",
    "triton==2.2.0",
    "transformers==4.41.2",
    "accelerate==0.31.0",
    "omegaconf==2.3.0",
    "torch==2.2.2",
    "torchvision==0.17.2",    
    "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@8d8ff45863416484b5b4bc547782591bbdfc696a#subdirectory=pipelines",
    "huggingface-hub==0.25.2",
    "oneflow",
    "setuptools>=75.2.0",
    "bitsandbytes>=0.44.1",
    "stable-fast",
    "tomesd>=0.1.3",
]

[tool.edge-maxxing]
models = [
    "stablediffusionapi/newdream-sdxl-20",
    "RobertML/cached-pipe-02"
]

[tool.uv.sources]
oneflow = { url = "https://github.com/siliconflow/oneflow_releases/releases/download/community_cu118/oneflow-0.9.1.dev20240802%2Bcu118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
stable-fast = { url = "https://github.com/chengzeyi/stable-fast/releases/download/v1.0.5/stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl" }

[project.scripts]
start_inference = "main:main"