Skip to content

vllm_omni.diffusion.utils.hf_utils ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

is_diffusion_model `cached` ¶

is_diffusion_model(model_name: str) -> bool

Check if a model is a diffusion model.

Uses multiple fallback strategies to detect diffusion models: 1. Check local file system for model_index.json (fastest, no imports) 2. Check using vllm's get_hf_file_to_dict utility 3. Try the standard diffusers approach (may fail due to import issues)

load_diffusers_config ¶

load_diffusers_config(model_name) -> dict