Update quantization.py

2023-04-24 21:30:06 +08:00
parent 5936557949
commit 38edd98341
1 changed files with 1 additions and 1 deletions
@@ -19,7 +19,7 @@ def find_layers(module, layers=[nn.Conv2d, nn.Linear], name=''):
 try:
    import triton
    import triton.language as tl
-    from .custom_autotune import *
+    from models.custom_autotune import *
 except:
    print('triton not installed. Run `pip install triton` to load quantized version of MOSS.')