o ٷi @snddlZddlmZddlmZddlmZeeZde fddZ de de fd d Z ede de fd d ZdS) N) lru_cache) init_logger)get_hf_file_to_dictreturncCsddlm}||}|S)Nr)DiffusionPipeline)"diffusers.pipelines.pipeline_utilsr load_config) model_namerconfigr V/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/diffusion/utils/hf_utils.pyload_diffusers_config s  r r cCsPztd|}|d}|dkrWdS|dpg}d|vWSty'YdSw)zABest-effort detection for Bagel (non-diffusers) diffusion models.z config.json model_typebagelT architecturesBagelForConditionalGenerationF)rget Exception)r cfgrrr r r _looks_like_bagels    rc Cstj|r]tj|d}tj|r]z1ddl}t| }||}Wdn1s,wY|drC|drCt dWdSWnt y\}z t d|WYd}~nd}~wwzt d|}|dru|drut d WdSWnt y}z t d |WYd}~nd}~wwzt |WdSttfy}zt d |t d WYd}~t|Sd}~wt y}zt d |WYd}~t|Sd}~ww)a9Check 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) zmodel_index.jsonrN _class_name_diffusers_versionz3Detected diffusion model via local model_index.jsonTz)Failed to read local model_index.json: %sz-Detected diffusion model via model_index.jsonzs