Alexander Bagus commited on
Commit
cca610a
·
1 Parent(s): ed4baf8
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ from videox_fun.pipeline import ZImageControlPipeline
18
  from videox_fun.models import ZImageControlTransformer2DModel
19
 
20
  repo_utils.clone_repo_if_not_exists("https://huggingface.co/Tongyi-MAI/Z-Image-Turbo", "app/models")
21
- repo_utils.clone_repo_if_not_exists("https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union", "app/models")
22
 
23
 
24
  # MODEL_REPO = "Tongyi-MAI/Z-Image-Turbo"
@@ -28,7 +28,7 @@ MAX_IMAGE_SIZE = 1280
28
  # git clone https://huggingface.co/Tongyi-MAI/Z-Image-Turbo
29
  MODEL_LOCAL = "models/Z-Image-Turbo/"
30
  # 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
31
- TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union/Z-Image-Turbo-Fun-Controlnet-Union.safetensors"
32
 
33
  weight_dtype = torch.bfloat16
34
 
 
18
  from videox_fun.models import ZImageControlTransformer2DModel
19
 
20
  repo_utils.clone_repo_if_not_exists("https://huggingface.co/Tongyi-MAI/Z-Image-Turbo", "app/models")
21
+ repo_utils.clone_repo_if_not_exists("https://huggingface.co/alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.0", "app/models")
22
 
23
 
24
  # MODEL_REPO = "Tongyi-MAI/Z-Image-Turbo"
 
28
  # git clone https://huggingface.co/Tongyi-MAI/Z-Image-Turbo
29
  MODEL_LOCAL = "models/Z-Image-Turbo/"
30
  # 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
31
+ TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union/Z-Image-Turbo-Fun-Controlnet-Union-2.0.safetensors"
32
 
33
  weight_dtype = torch.bfloat16
34