umer1995 commited on
Commit
957aeac
·
verified ·
1 Parent(s): 4f4dc6c

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

Browse files
Files changed (1) hide show
  1. app.py +9 -2
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 and "fantasytalking" not in text
 
 
37
  ):
38
  models_init.write_text(
39
- "from transformers import Mistral3ForConditionalGeneration, PixtralProcessor\n"
 
 
 
 
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"