Deploy NFA Track R FLUX.2 Fun CN ZeroGPU (real depth CN)
Browse files
app.py
CHANGED
|
@@ -33,15 +33,22 @@ def _patch_videox_models_init(root: Path) -> None:
|
|
| 33 |
if models_init.is_file():
|
| 34 |
text = models_init.read_text(encoding="utf-8", errors="replace")
|
| 35 |
if not (
|
| 36 |
-
"Flux2ControlTransformer2DModel" in text
|
|
|
|
|
|
|
| 37 |
):
|
| 38 |
models_init.write_text(
|
| 39 |
-
"from transformers import
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
"from .flux2_image_processor import Flux2ImageProcessor\n"
|
| 41 |
"from .flux2_transformer2d import Flux2Transformer2DModel\n"
|
| 42 |
"from .flux2_transformer2d_control import Flux2ControlTransformer2DModel\n"
|
| 43 |
"from .flux2_vae import AutoencoderKLFlux2\n"
|
| 44 |
"__all__ = [\n"
|
|
|
|
| 45 |
" 'AutoencoderKLFlux2',\n"
|
| 46 |
" 'Flux2ControlTransformer2DModel',\n"
|
| 47 |
" 'Flux2ImageProcessor',\n"
|
|
|
|
| 33 |
if models_init.is_file():
|
| 34 |
text = models_init.read_text(encoding="utf-8", errors="replace")
|
| 35 |
if not (
|
| 36 |
+
"Flux2ControlTransformer2DModel" in text
|
| 37 |
+
and "fantasytalking" not in text
|
| 38 |
+
and "AutoProcessor" in text
|
| 39 |
):
|
| 40 |
models_init.write_text(
|
| 41 |
+
"from transformers import (\n"
|
| 42 |
+
" AutoProcessor,\n"
|
| 43 |
+
" Mistral3ForConditionalGeneration,\n"
|
| 44 |
+
" PixtralProcessor,\n"
|
| 45 |
+
")\n"
|
| 46 |
"from .flux2_image_processor import Flux2ImageProcessor\n"
|
| 47 |
"from .flux2_transformer2d import Flux2Transformer2DModel\n"
|
| 48 |
"from .flux2_transformer2d_control import Flux2ControlTransformer2DModel\n"
|
| 49 |
"from .flux2_vae import AutoencoderKLFlux2\n"
|
| 50 |
"__all__ = [\n"
|
| 51 |
+
" 'AutoProcessor',\n"
|
| 52 |
" 'AutoencoderKLFlux2',\n"
|
| 53 |
" 'Flux2ControlTransformer2DModel',\n"
|
| 54 |
" 'Flux2ImageProcessor',\n"
|