Add detection for SD1.5 models (AbyssOrangeMix, NAI, etc.)

Models like Faber8/AbyssOrangeMix2 are SD1.5 models, not Flux.
Now detected as StableDiffusionPipeline instead of FluxPipeline.
parent 8082b88e
...@@ -8469,6 +8469,9 @@ def main(args): ...@@ -8469,6 +8469,9 @@ def main(args):
detected_class = "CogVideoXPipeline" detected_class = "CogVideoXPipeline"
elif "flux" in model_id_lower: elif "flux" in model_id_lower:
detected_class = "FluxPipeline" detected_class = "FluxPipeline"
# SD1.5 models (AbyssOrangeMix, etc.)
elif any(x in model_id_lower for x in ["abyssor", "abyss", "orangemix", "nai", "novelai", "deliberate", "dreamshaper", "realistic", "cyberrealistic"]):
detected_class = "StableDiffusionPipeline"
if detected_class and detected_class != m_info["class"]: if detected_class and detected_class != m_info["class"]:
print(f"\n⚠️ Pipeline component mismatch detected!") print(f"\n⚠️ Pipeline component mismatch detected!")
...@@ -9005,6 +9008,9 @@ def main(args): ...@@ -9005,6 +9008,9 @@ def main(args):
detected_class = "StableDiffusion3Pipeline" detected_class = "StableDiffusion3Pipeline"
elif "sd15" in img_model_id_lower or "stable-diffusion-1.5" in img_model_id_lower: elif "sd15" in img_model_id_lower or "stable-diffusion-1.5" in img_model_id_lower:
detected_class = "StableDiffusionPipeline" detected_class = "StableDiffusionPipeline"
# SD1.5 models (AbyssOrangeMix, etc.)
elif any(x in img_model_id_lower for x in ["abyssor", "abyss", "orangemix", "nai", "novelai", "deliberate", "dreamshaper", "realistic", "cyberrealistic"]):
detected_class = "StableDiffusionPipeline"
if detected_class and detected_class != img_info["class"]: if detected_class and detected_class != img_info["class"]:
print(f"\n⚠️ Image model pipeline component mismatch detected!") print(f"\n⚠️ Image model pipeline component mismatch detected!")
......
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