From b612127a8098732c6a08ad363cc01ac87ca7921f Mon Sep 17 00:00:00 2001
From: Vixy <vixy@k4zka.online>
Date: Thu, 1 Jan 2026 20:13:02 -0600
Subject: [PATCH] Fix onnxruntime for Jetson ARM64

---
 Dockerfile       | 4 ++++
 requirements.txt | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index ea8485c..0112222 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -27,6 +27,10 @@ COPY requirements.txt /app/
 # Note: torch and torchvision are already in the base image
 RUN pip3 install --no-cache-dir -r requirements.txt
 
+# Install onnxruntime for Jetson (from NVIDIA's pre-built wheels)
+# The base dustynv image should have it, but ensure it's available
+RUN pip3 install --no-cache-dir onnxruntime || echo "onnxruntime already installed or using system version"
+
 # Copy application code
 COPY config.py /app/
 COPY main.py /app/
diff --git a/requirements.txt b/requirements.txt
index 86db735..b20c555 100755
--- a/requirements.txt
+++ b/requirements.txt
@@ -23,7 +23,7 @@ opencv-python==4.9.0.80
 
 # IP-Adapter FaceID (for consistent face generation)
 insightface==0.7.3
-onnxruntime-gpu==1.17.0
+# onnxruntime-gpu installed separately for Jetson (from NVIDIA wheels)
 
 # Utilities
 pydantic==2.6.0