We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 10f841b commit 7a52475Copy full SHA for 7a52475
vllm_omni/diffusion/utils/hf_utils.py
@@ -18,13 +18,13 @@ def _looks_like_bagel(model_name: str) -> bool:
18
"""Best-effort detection for Bagel (non-diffusers) diffusion models."""
19
try:
20
cfg = get_hf_file_to_dict("config.json", model_name)
21
+ model_type = cfg.get("model_type")
22
+ if model_type == "bagel":
23
+ return True
24
+ architectures = cfg.get("architectures") or []
25
+ return "BagelForConditionalGeneration" in architectures
26
except Exception:
27
return False
- model_type = cfg.get("model_type")
- if model_type == "bagel":
- return True
- architectures = cfg.get("architectures") or []
- return "BagelForConditionalGeneration" in architectures
28
29
30
@lru_cache
0 commit comments