repo_id: "DenisKochetov/Comfy3D_Pre_Builds"
supported_cuda_versions: ["12.8", "12.6", "12.4", "12.1", "11.8"]
cuda_version: "12.8"

# Packages inside this file will be pip downloaded before actual build start
build_requirements: "build_requirements.txt"
# Dependencies to build wheels from
dependencies: "dependencies.txt"
# Folder used to store `git clone` project files from dependencies
dependencies_dir_name: "_Build_Dependencies"
# Folder used to store library files download from remote
libs_dir_name: "_Libs"
# Folder used to store built wheels
wheels_dir_name: "_Build_Wheels"

build_base_packages: ["torch", "torchvision", "torchaudio", "xformers"]

# Packages requiring special install flags (like --no-build-isolation)
isolated_packages: ["diso"]

# Platform-specific packages
platform_packages:
  win:
    - "triton-windows<3.4"
  linux:
    - "triton"

# Simple spconv installation with CUDA mapping
spconv_cuda_mapping:
  "12.8": "cu126"  # Default for 12.8 and maximum available
  "12.6": "cu126"
  "12.4": "cu124"
  "12.1": "cu121"
  "11.8": "cu118"

# All packages will be automatically installed at beginning of install.py follwing orders from top to bottom 
remote_packages:
  torch:
    version: "2.7.0"
    url: "https://download.pytorch.org/whl/${cuda_version}"
  
  torchvision:
    version: "0.22.0"
    url: "https://download.pytorch.org/whl/${cuda_version}"

  torchaudio:
    version: "2.7.0"
    url: "https://download.pytorch.org/whl/${cuda_version}"

  xformers:
    version: "0.0.30"
    url: "https://download.pytorch.org/whl/${cuda_version}"

  torch-scatter:
   url_option: "--find-links"
   url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html"
  
  diso:
    url: "git+https://github.com/SarahWeiii/diso.git#egg=diso"
    install_flags: ["--no-build-isolation"]