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

app.py

@@ -30,28 +30,39 @@ _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)
+    if models_init.is_file():
+        text = models_init.read_text(encoding="utf-8", errors="replace")
+        if not (
+            "Flux2ControlTransformer2DModel" in text and "fantasytalking" not in text
+        ):
+            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)
+
+    pipe_init = root / "videox_fun" / "pipeline" / "__init__.py"
+    if pipe_init.is_file():
+        text = pipe_init.read_text(encoding="utf-8", errors="replace")
+        if "pipeline_cogvideox" in text or "Flux2ControlPipeline" not in text:
+            pipe_init.write_text(
+                "from .pipeline_flux2_control import Flux2ControlPipeline\n"
+                "__all__ = ['Flux2ControlPipeline']\n",
+                encoding="utf-8",
+            )
+            print("[nfa-fun-cn] patched videox_fun.pipeline.__init__ (Flux2-only)", flush=True)
 
 
 def _ensure_videox_on_path() -> None: