diff --git a/artifact_comfyui_download.sh b/artifact_comfyui_download.sh
index 3f8cd71..22623da 100755
--- a/artifact_comfyui_download.sh
+++ b/artifact_comfyui_download.sh
@@ -460,7 +460,7 @@ link_model() {
         # Look for checkpoint .safetensors files in the snapshot root (not in subdirectories)
         # These are the consolidated model files like sd_xl_base_1.0.safetensors
         local checkpoint_files
-        checkpoint_files=$(echo "$model_files" | grep -v '/' | grep '\.safetensors$' || echo "$model_files" | grep -E 'snapshots/[^/]+/[^/]+\.safetensors$' | grep -vE '/(vae|text_encoder|transformer|unet)/' || true)
+        checkpoint_files=$(echo "$model_files" | grep -E 'snapshots/[^/]+/[^/]+\.safetensors$' | grep -vE '/(vae|text_encoder|transformer|unet|tokenizer)/' || true)
 
         if [[ -n "$checkpoint_files" ]]; then
             mkdir -p "$checkpoint_dir"