[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:
1 parent
0e07a0ae9e
commit
f73791d246
1 file changed
+2
-1
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user