21 lines
464 B
Python
21 lines
464 B
Python
from functools import lru_cache
|
|
|
|
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
|
|
from vllm.logger import init_logger
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
def load_diffusers_config(model_name) -> dict:
|
|
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
|