Deploy NFA Track R FLUX.2 Fun CN ZeroGPU (real depth CN)

app.py

@@ -28,9 +28,36 @@ _VX = APP_DIR / "vendor" / "VideoX-Fun"
 _VX_CACHE = Path.home() / "VideoX-Fun"
 
 
+def _patch_videox_models_init(root: Path) -> None:
+    models_init = root / "videox_fun" / "models" / "__init__.py"
+    if not models_init.is_file():
+        return
+    text = models_init.read_text(encoding="utf-8", errors="replace")
+    if "Flux2ControlTransformer2DModel" in text and "fantasytalking" not in text:
+        return
+    models_init.write_text(
+        "from transformers import Mistral3ForConditionalGeneration, PixtralProcessor\n"
+        "from .flux2_image_processor import Flux2ImageProcessor\n"
+        "from .flux2_transformer2d import Flux2Transformer2DModel\n"
+        "from .flux2_transformer2d_control import Flux2ControlTransformer2DModel\n"
+        "from .flux2_vae import AutoencoderKLFlux2\n"
+        "__all__ = [\n"
+        "    'AutoencoderKLFlux2',\n"
+        "    'Flux2ControlTransformer2DModel',\n"
+        "    'Flux2ImageProcessor',\n"
+        "    'Flux2Transformer2DModel',\n"
+        "    'Mistral3ForConditionalGeneration',\n"
+        "    'PixtralProcessor',\n"
+        "]\n",
+        encoding="utf-8",
+    )
+    print("[nfa-fun-cn] patched videox_fun.models.__init__ (Flux2-only)", flush=True)
+
+
 def _ensure_videox_on_path() -> None:
     for candidate in (_VX, _VX_CACHE):
         if (candidate / "videox_fun" / "models").is_dir():
+            _patch_videox_models_init(candidate)
             p = str(candidate)
             if p not in sys.path:
                 sys.path.insert(0, p)
@@ -53,25 +80,7 @@ def _ensure_videox_on_path() -> None:
             str(_VX_CACHE),
         ]
     )
-    # Replace heavy models/__init__.py so Flux2 Fun CN imports do not require
-    # the full audio/video dependency stack.
-    models_init = _VX_CACHE / "videox_fun" / "models" / "__init__.py"
-    models_init.write_text(
-        "from transformers import Mistral3ForConditionalGeneration, PixtralProcessor\n"
-        "from .flux2_image_processor import Flux2ImageProcessor\n"
-        "from .flux2_transformer2d import Flux2Transformer2DModel\n"
-        "from .flux2_transformer2d_control import Flux2ControlTransformer2DModel\n"
-        "from .flux2_vae import AutoencoderKLFlux2\n"
-        "__all__ = [\n"
-        "    'AutoencoderKLFlux2',\n"
-        "    'Flux2ControlTransformer2DModel',\n"
-        "    'Flux2ImageProcessor',\n"
-        "    'Flux2Transformer2DModel',\n"
-        "    'Mistral3ForConditionalGeneration',\n"
-        "    'PixtralProcessor',\n"
-        "]\n",
-        encoding="utf-8",
-    )
+    _patch_videox_models_init(_VX_CACHE)
     sys.path.insert(0, str(_VX_CACHE))