From 399904bf0e0b89fcd23a8b5190eaa863da21a898 Mon Sep 17 00:00:00 2001 From: CSY-ModelCloud Date: Wed, 13 May 2026 15:36:46 +0800 Subject: [PATCH] fix AttributeError: 'FakeGPTQModel' object has no attribute '_sanitize_input_ids_for_embeddings' --- tests/test_calibration_data_device.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_calibration_data_device.py b/tests/test_calibration_data_device.py index 6a75822f6..d8dfcd378 100644 --- a/tests/test_calibration_data_device.py +++ b/tests/test_calibration_data_device.py @@ -16,13 +16,11 @@ """ import os -import types -import unittest - import pytest import torch import torch.nn as nn - +import types +import unittest from gptqmodel.models.base import BaseQModel from gptqmodel.quantization import QuantizeConfig @@ -210,6 +208,7 @@ class FakeGPTQModel: move_input_capture_example = BaseQModel.move_input_capture_example prepare_layer_replay_kwargs = BaseQModel.prepare_layer_replay_kwargs run_input_capture = BaseQModel.run_input_capture + _sanitize_input_ids_for_embeddings = BaseQModel._sanitize_input_ids_for_embeddings def __init__(self): self.quantize_config = types.SimpleNamespace( @@ -355,6 +354,7 @@ class FakeGPTQModel: move_input_capture_example = BaseQModel.move_input_capture_example prepare_layer_replay_kwargs = BaseQModel.prepare_layer_replay_kwargs run_input_capture = BaseQModel.run_input_capture + _sanitize_input_ids_for_embeddings = BaseQModel._sanitize_input_ids_for_embeddings def __init__(self): self.quantize_config = types.SimpleNamespace( @@ -509,6 +509,7 @@ class FakeGPTQModel: move_input_capture_example = BaseQModel.move_input_capture_example prepare_layer_replay_kwargs = BaseQModel.prepare_layer_replay_kwargs run_input_capture = BaseQModel.run_input_capture + _sanitize_input_ids_for_embeddings = BaseQModel._sanitize_input_ids_for_embeddings def __init__(self): self.quantize_config = types.SimpleNamespace(