mirror of https://github.com/THUDM/CodeGeeX.git
Merge pull request #128 from makenv/fix-oneflow-quantize
避免quantization在缺少oneflow模块的时候出错.pull/172/head
commit
494219a43a
@ -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
|
||||
|
Loading…
Reference in New Issue