From 38edd98341cc5c80042ab3772a5f69bc616832a4 Mon Sep 17 00:00:00 2001
From: Hzfinfdu <74953051+Hzfinfdu@users.noreply.github.com>
Date: Mon, 24 Apr 2023 21:30:06 +0800
Subject: [PATCH] Update quantization.py

---
 models/quantization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/quantization.py b/models/quantization.py
index c4e2a22..38e89d7 100644
--- a/models/quantization.py
+++ b/models/quantization.py
@@ -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.')