Validate base model exists before adding LoRA to model list

- When --update-models detects a LoRA adapter, validate that the base
  model exists on HuggingFace before adding it to the model list
- Skip LoRAs whose base models are not found on HuggingFace
- Added support for flux and sdxl base model detection
- Print informative messages when skipping LoRAs with missing base models
parent 2e8b5bc7
......@@ -1367,6 +1367,19 @@ def update_all_models(hf_token=None):
base_model = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
elif "svd" in model_id.lower() or "stable-video" in model_id.lower():
base_model = "stabilityai/stable-video-diffusion-img2vid-xt-1-1"
elif "flux" in model_id.lower():
base_model = "black-forest-labs/FLUX.1-dev"
elif "sdxl" in model_id.lower() or "xl" in model_id.lower():
base_model = "stabilityai/stable-diffusion-xl-base-1.0"
# Validate that the base model exists on HuggingFace
if base_model:
base_model_info = validate_hf_model(base_model, hf_token=hf_token)
if not base_model_info:
print(f" ⏭️ Skipping LoRA {model_id} - base model not found: {base_model}")
continue
else:
print(f" 📦 LoRA detected: {model_id} (base: {base_model})")
# Build model entry
model_entry = {
......@@ -1455,6 +1468,19 @@ def update_all_models(hf_token=None):
base_model = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
elif "svd" in model_id.lower() or "stable-video" in model_id.lower():
base_model = "stabilityai/stable-video-diffusion-img2vid-xt-1-1"
elif "flux" in model_id.lower():
base_model = "black-forest-labs/FLUX.1-dev"
elif "sdxl" in model_id.lower() or "xl" in model_id.lower():
base_model = "stabilityai/stable-diffusion-xl-base-1.0"
# Validate that the base model exists on HuggingFace
if base_model:
base_model_info = validate_hf_model(base_model, hf_token=hf_token)
if not base_model_info:
print(f" ⏭️ Skipping LoRA {model_id} - base model not found: {base_model}")
continue
else:
print(f" 📦 LoRA detected: {model_id} (base: {base_model})")
# Build model entry
model_entry = {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment