22

app.py

@@ -24,7 +24,7 @@ repo_utils.clone_repo_if_not_exists("https://huggingface.co/alibaba-pai/Z-Image-
 
 MODEL_LOCAL = "models/Z-Image-Turbo/"
 TRANSFORMER_LOCAL = "models/Z-Image-Turbo-Fun-Controlnet-Union-2.0/Z-Image-Turbo-Fun-Controlnet-Union-2.0.safetensors"
-TRANSFORMER_CONFIG = "~/app/config/z_image/z_image_control_2.0.yaml"
+TRANSFORMER_CONFIG = repo_utils.convert_path("config/z_image/z_image_control_2.0.yaml")
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1280

utils/repo_utils.py

@@ -52,9 +52,8 @@ def move_folder(source_path, destination_path):
 
 
 def convert_path(mypath):
-    home_dir = os.path.expanduser("~")
-    # app_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
-    converted_path = os.path.join(home_dir, mypath)
+    app_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
+    converted_path = os.path.join(app_dir, mypath)
     print(mypath)
     print(converted_path)
     return converted_path