diff --git a/artifact_comfyui_download.sh b/artifact_comfyui_download.sh
index ea4d3a0..54689c0 100755
--- a/artifact_comfyui_download.sh
+++ b/artifact_comfyui_download.sh
@@ -310,14 +310,14 @@ find_model_files() {
     local repo_id="$1"
     local filename_filter="$2"
 
-    python3 - <<EOPYFINDFIND
+    python3 <<EOPYFINDFIND - "$CACHE_DIR" "$repo_id" "$filename_filter"
 import os
 import sys
 from pathlib import Path
 
-cache_dir = '${CACHE_DIR}'
-repo_id = '${repo_id}'
-filename_filter = '${filename_filter}'
+cache_dir = sys.argv[1]
+repo_id = sys.argv[2]
+filename_filter = sys.argv[3]
 
 # HuggingFace cache structure: cache_dir/models--org--name/snapshots/hash/
 # Try both with and without 'hub/' subdirectory for compatibility