fix: improve checkpoint file detection pattern for SDXL models

This commit is contained in:
2025-11-22 16:46:30 +01:00
parent 12be347d35
commit 11b3d48c9b

View File

@@ -460,7 +460,7 @@ link_model() {
# Look for checkpoint .safetensors files in the snapshot root (not in subdirectories) # 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 # These are the consolidated model files like sd_xl_base_1.0.safetensors
local checkpoint_files 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 if [[ -n "$checkpoint_files" ]]; then
mkdir -p "$checkpoint_dir" mkdir -p "$checkpoint_dir"