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 = "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

@@ -50,3 +50,8 @@ def move_folder(source_path, destination_path):
     except Exception as e:
         print(f"Error move_folder: {e}")
 
+
+def convert_path(mypath):
+    app_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
+    converted_path = os.path.join(app_dir, mypath)
+    return converted_path