knlnguyen1802andn00909098
f73791d246
Signed-off-by: n00909098 <nguyen.kha.long@huawei.com> Co-authored-by: n00909098 <nguyen.kha.long@huawei.com>
22 lines
469 B
Python
22 lines
469 B
Python
from functools import lru_cache
|
|
|
|
from vllm.logger import init_logger
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
def load_diffusers_config(model_name) -> dict:
|
|
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
|
|
|
config = DiffusionPipeline.load_config(model_name)
|
|
return config
|
|
|
|
|
|
@lru_cache
|
|
def is_diffusion_model(model_name: str) -> bool:
|
|
try:
|
|
load_diffusers_config(model_name)
|
|
return True
|
|
except Exception:
|
|
return False
|