避免quantization在缺少oneflow模块的时候出错.

2 years ago · 952ae590b0
parent f6636b728a
commit 952ae590b0
1 changed files with 5 additions and 2 deletions
--- a/codegeex/quantization/init.py
+++ b/codegeex/quantization/init.py
@ -1,3 +1,6 @@
 from .quantize import quantize
 try:
    from .quantize_oneflow import quantize_oneflow
    from .quantize_oneflow import QuantizedLinear
 except ModuleNotFoundError:
    pass