Skip to content

Commit 6345f86

Browse files
committed
added ark folder & removed qbits
1 parent b1b60e4 commit 6345f86

File tree

5 files changed

+420
-697
lines changed

5 files changed

+420
-697
lines changed

auto_round_extension/qbits/__init__.py renamed to auto_round_extension/ark/__init__.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from auto_round_extension.qbits.qlinear_qbits import QuantLinear as QBitsQuantLinear
16-
from auto_round_extension.qbits.qlinear_qbits_gptq import (
17-
QuantLinear as QBitsGPTQQuantLinear,
18-
)
19-
from auto_round_extension.qbits.qbits_awq import QuantLinear as QBitsAWQQuantLinear
15+
from auto_round_extension.ark.qlinear import QuantLinear, QuantLinearGPTQ, QuantLinearAWQ
16+
qlinear_classes = (QuantLinear, QuantLinearGPTQ)
2017

21-
qbits_qlinear_classes = (QBitsQuantLinear, QBitsGPTQQuantLinear)
22-
23-
qbits_awq_classes = (QBitsAWQQuantLinear,)
18+
awq_classes = (QuantLinearAWQ,)

0 commit comments

Comments
 (0)