22

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import numpy as np
 import random, json, spaces, torch, time
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
-from videox_fun.pipeline import ZImageControlPipeline
-from videox_fun.models import ZImageControlTransformer2DModel
+
 from transformers import AutoTokenizer, Qwen3ForCausalLM
 from diffusers import AutoencoderKL
 from utils.image_utils import get_image_latent, rescale_image
@@ -12,11 +11,17 @@ from utils import repo_utils
 # from controlnet_aux import HEDdetector, MLSDdetector, OpenposeDetector, CannyDetector, MidasDetector
 from controlnet_aux.processor import Processor
 
+# clone and move videox_fun
+repo_utils.clone_repo_if_not_exists("https://github.com/aigc-apps/VideoX-Fun.git", "app/repos")
+repo_utils.move_folder("app/repos/VideoX-Fun/videox_fun", "app/videox_fun")
+from videox_fun.pipeline import ZImageControlPipeline
+from videox_fun.models import ZImageControlTransformer2DModel
 
 repo_utils.clone_repo_if_not_exists("https://huggingface.co/Tongyi-MAI/Z-Image-Turbo", "app/models")
 repo_utils.clone_repo_if_not_exists("https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union", "app/models")
 
 
+
 # MODEL_REPO = "Tongyi-MAI/Z-Image-Turbo"
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1280

repos/VideoX-Fun

@@ -0,0 +1 @@
+Subproject commit 4935a7039153498b1a8bc9bce06ab8bc99d94f6e

utils/repo_utils.py

@@ -1,4 +1,4 @@
-import os, subprocess
+import os, subprocess, shutil
 
 def clone_repo_if_not_exists(git_url, git_dir):
     """
@@ -31,3 +31,16 @@ def clone_repo_if_not_exists(git_url, git_dir):
     else:
         print(f"Repository '{git_name}' already exists at '{repo_path}'. Skipping clone.")
 
+
+def move_folder(source_path, destination_path):
+    """
+    Moves a folder from source_path to destination_path.
+    If a folder already exists at destination_path, it will be replaced.
+    """
+    home_dir = os.path.expanduser("~")
+    source_path = os.path.join(home_dir, source_path)
+    destination_path = os.path.join(home_dir, destination_path)
+
+    if shutil.os.path.exists(destination_path):
+        shutil.rmtree(destination_path)  # Remove existing destination folder
+    shutil.move(source_path, destination_path)