CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-06-04 21:05:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update 30b req

Browse Source
This commit is contained in:
maybleMyers 2025-12-08 14:44:56 -08:00
parent d8c88bc223
commit b8f6c4ecbc
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
vlm.py
View File

@ -123,6 +123,11 @@ class VLMManager:
with open(config_path, "r", encoding="utf-8") as f:
config = json.load(f)
model_type = config.get("model_type", "").lower()
# Check for MoE models first
if "moe" in model_type:
if "qwen3" in model_type:
return "qwen3_vl_moe"
# Then standard models
if "qwen3" in model_type:
return "qwen3_vl"
elif "qwen2_5" in model_type or "qwen2.5" in model_type:
@ -185,7 +190,9 @@ class VLMManager:
progress(0.5, desc=f"Loading model weights ({model_type})...")
# Select the correct model class based on detected type
if model_type == "qwen3_vl":
if model_type == "qwen3_vl_moe":
from transformers import Qwen3VLMoeForConditionalGeneration as ModelClass
elif model_type == "qwen3_vl":
from transformers import Qwen3VLForConditionalGeneration as ModelClass
else:
from transformers import Qwen2_5_VLForConditionalGeneration as ModelClass