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

kester2015 · Apr 21, 2023 · 952ae59 · 952ae59
1 parent f6636b7
commit 952ae59
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