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

2023-04-21 18:16:31 +08:00 · 2023-04-21 18:16:31 +08:00 · 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
-from .quantize_oneflow import quantize_oneflow
-from .quantize_oneflow import QuantizedLinear
+try:
+    from .quantize_oneflow import quantize_oneflow
+    from .quantize_oneflow import QuantizedLinear
+except ModuleNotFoundError:
+    pass