Merge pull request THUDM#128 from makenv/fix-oneflow-quantize

避免quantization在缺少oneflow模块的时候出错.
WenTianShi · Apr 22, 2023 · 494219a · 494219a
2 parents e824af4 + 952ae59
commit 494219a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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