[CI] Patch torch.library.infer_schema for fused moe ops to fix CI (#854)
make sure pytorch infer_schema check is patched before some case which using fused moe ops: 1. model register 2. quantization loading 3. fused moe ut Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -23,5 +23,9 @@ def register():
|
||||
|
||||
|
||||
def register_model():
|
||||
# fix pytorch schema check error, remove this line after pytorch
|
||||
# is upgraded to 2.7.0
|
||||
import vllm_ascend.patch.worker.patch_common.patch_utils # noqa: F401
|
||||
|
||||
from .models import register_model
|
||||
register_model()
|
||||
|
||||
Reference in New Issue
Block a user