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

[project]
name = "flux-schnell-edge-inference"
description = "An edge-maxxing model submission by RobertML for the 4090 Flux contest"
requires-python = ">=3.10,<3.13"
version = "8"
dependencies = [
  "diffusers==0.31.0",
  "transformers==4.46.2",
  "accelerate==1.1.0",
  "omegaconf==2.3.0",
  "torch==2.5.1",
  "protobuf==5.28.3",
  "sentencepiece==0.2.0",
  "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@7c760ac54f6052803dadb3ade8ebfc9679a94589#subdirectory=pipelines",
  "gitpython>=3.1.43",
  "hf_transfer==0.1.8",
  "torchao==0.6.1",
  "setuptools"
]

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra0Chro0"
revision = "197fee33effc13f33a63b44e4ed33fd8789adcab"
exclude = ["transformer"]

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra1Chro1"
revision = "e0ec520a34977320b1f42026f7a5fd008a695379"

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra2Chro2"
revision = "0c4fe40f12dc7cea37f69d8ed8e5f253e27db2e6"

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra3Chro3"
revision = "4e9f28cc4aff66d8138836e723f8659760d80ef2"


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