diff --git a/quantization.py b/quantization.py index 42836a1..bf30790 100644 --- a/quantization.py +++ b/quantization.py @@ -7,11 +7,7 @@ import base64 import ctypes from typing import List - -try: - from cpm_kernels.kernels.base import LazyKernelCModule, KernelFunction, round_up -except: - pass +from cpm_kernels.kernels.base import LazyKernelCModule, KernelFunction, round_up class W8A16Linear(torch.autograd.Function):