From 38723ebb07d2e98a4464b1fdb8217ae63b30abd3 Mon Sep 17 00:00:00 2001
From: Peter Schroedl <peter_schroedl@me.com>
Date: Thu, 5 Dec 2024 23:37:43 +0100
Subject: [PATCH] add ARCH_LIST and install package for cuda extensions

---
 requirements.txt | 2 +-
 setup.py         | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index bcc6074..d66db70 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,4 @@ tqdm>=4.66.1
 hydra-core>=1.3.2
 iopath>=0.1.10
 pillow>=9.4.0
--e .
+git+https://github.com/pschroedl/ComfyUI-SAM2-Realtime.git@main#egg=sam2_realtime
diff --git a/setup.py b/setup.py
index 3b58a7a..583e930 100644
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,10 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
+import os
+# Set the CUDA architecture list
+os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0 8.6+PTX 8.7 9.0 9.0a"
+
 from setuptools import find_packages, setup
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
 
@@ -53,7 +57,7 @@ setup(
     license=LICENSE,
     packages=find_packages(),
     install_requires=REQUIRED_PACKAGES,
-    python_requires=">=3.10",
+    python_requires=">=3.10.15",
     ext_modules=get_extensions(),
     cmdclass={"build_ext": BuildExtension.with_options(no_python_abi_suffix=True)},
 )