Perception-LM-1B-Int4-NotBNB / modeling_perception_lm_int4.py
Dhruvil03's picture
Upload INT4 quantized Perception-LM-1B
5db39a7 verified
from transformers.models.perception_lm.modeling_perception_lm import PerceptionLMForConditionalGeneration
from .int4_quant import quantize_model_to_int4
class PerceptionLMForConditionalGenerationInt4(PerceptionLMForConditionalGeneration):
def __init__(self, config):
super().__init__(config)
exclude = getattr(config, "int4_exclude_patterns", [])
quantize_model_to_int4(self, name_exclude_patterns=tuple(exclude))