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

[project]
name = "flux-schnell-edge-inference"
description = "An edge-maxxing model submission 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",
  "torchao==0.6.1",
  "hf_transfer==0.1.8",
  "setuptools==75.2.0",
  "edge-maxxing-pipelines @ git+https://github.com/womboai/edge-maxxing@7c760ac54f6052803dadb3ade8ebfc9679a94589#subdirectory=pipelines",
]

[[tool.edge-maxxing.models]]
repository = "black-forest-labs/FLUX.1-schnell"
revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
exclude = ["transformer", "vae", "text_encoder_2"]

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra0Jan10"
revision = "d3ded25a77fdef06de4059d94b080a34da6e7a82"

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra1Jan11"
revision = "c76831ddf0852be22835f79dc5c1fbacb1ccda9e"

[[tool.edge-maxxing.models]]
repository = "TrendForge/extra2Jan12"
revision = "da7c5cf904a9dbba65a7282396befa49623cd9cd"

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