Spaces:

AiSudo
/

ZIT-Controlnet

Running on Zero

Alexander Bagus commited on 1 day ago

Commit

cca610a

1 Parent(s): ed4baf8

22

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ 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"
@@ -28,7 +28,7 @@ MAX_IMAGE_SIZE = 1280
 # git clone https://huggingface.co/Tongyi-MAI/Z-Image-Turbo
 MODEL_LOCAL = "models/Z-Image-Turbo/"
 # curl -L -o Z-Image-Turbo-Fun-Controlnet-Union.safetensors https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union/resolve/main/Z-Image-Turbo-Fun-Controlnet-Union.safetensors
-TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union/Z-Image-Turbo-Fun-Controlnet-Union.safetensors"
 weight_dtype = torch.bfloat16

 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-2.0", "app/models")
 # MODEL_REPO = "Tongyi-MAI/Z-Image-Turbo"
 # git clone https://huggingface.co/Tongyi-MAI/Z-Image-Turbo
 MODEL_LOCAL = "models/Z-Image-Turbo/"
 # curl -L -o Z-Image-Turbo-Fun-Controlnet-Union.safetensors https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union/resolve/main/Z-Image-Turbo-Fun-Controlnet-Union.safetensors
+TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union/Z-Image-Turbo-Fun-Controlnet-Union-2.0.safetensors"
 weight_dtype = torch.bfloat16