github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-06 03:20:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

[HotFix] CI,import,requirements-test for #5838 (#5892)

Browse Source 
* [Hot Fix] CI,import,requirements-test

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Runyu Lu
2024-07-08 22:32:06 +08:00
committed by GitHub
parent cba20525a8
commit 66abf1c6e8
4 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/inference/utils.py
View File

@@ -186,8 +186,6 @@ def get_model_type(model_or_path: Union[nn.Module, str, DiffusionPipeline]):
"""
try:
from diffusers import DiffusionPipeline
DiffusionPipeline.load_config(model_or_path)
return ModelType.DIFFUSION_MODEL
except: