fix: improve checkpoint file detection pattern for SDXL models
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user