Update quant.py

quant.py

@@ -120,7 +120,8 @@ class Quantizer(nn.Module):
 
 
 try:
-    import quant_cuda
+    import importlib
+    quant_cuda = importlib.import_module("quant_cuda")
 except:
     import os
     import sys