mirror of https://github.com/THUDM/CodeGeeX.git
避免quantization在缺少oneflow模块的时候出错.
parent
f6636b728a
commit
952ae590b0
@ -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