[Bugfix] Init early toch cuda (#185)

Signed-off-by: n00909098 <nguyen.kha.long@huawei.com>
Co-authored-by: n00909098 <nguyen.kha.long@huawei.com>
This commit is contained in:
knlnguyen1802andn00909098 authored and GitHub committed 2025-12-04 11:38:52 +08:00
1 parent 0e07a0ae9e
commit f73791d246
1 file changed
+2 -1
+2 -1
View File
@@ -1,12 +1,13 @@
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:
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
config = DiffusionPipeline.load_config(model_name)
return config