From 0f89c6a6e2d3b2f9935d5f71302d945c6e2aed68 Mon Sep 17 00:00:00 2001
From: huchukato <huchukato@gmail.com>
Date: Fri, 27 Feb 2026 10:13:50 +0100
Subject: [PATCH] Simplify auto-installation logic - remove complex dummy
 module system that was causing issues

---
 __init__.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/__init__.py b/__init__.py
index e276b5f..5b5fbd2 100644
--- a/__init__.py
+++ b/__init__.py
@@ -81,6 +81,12 @@ def detect_cloud_environment():
 # Auto-detect CUDA and install appropriate TensorRT packages
 def _auto_install_tensorrt():
     """Auto-detect CUDA version and install appropriate TensorRT packages if needed"""
+    # Check if auto-install is disabled
+    disable_auto_install = os.environ.get("DISABLE_TENSORRT_AUTO_INSTALL", "false").lower() == "true"
+    if disable_auto_install:
+        print("[ComfyUI-RIFE-TensorRT] Auto-installation disabled via DISABLE_TENSORRT_AUTO_INSTALL")
+        return True
+    
     try:
         # Check if TensorRT is already installed
         try: