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)
|
# 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"
|
||||||
|
|||||||
Reference in New Issue
Block a user