|
15 | 15 | import copy
|
16 | 16 | import inspect
|
17 | 17 | import logging
|
18 |
| -import warnings |
19 | 18 | from enum import Enum
|
20 | 19 | from itertools import chain
|
21 | 20 | from pathlib import Path
|
|
31 | 30 | from neural_compressor.quantization import fit
|
32 | 31 | from torch.utils.data import DataLoader, RandomSampler
|
33 | 32 | from transformers import (
|
34 |
| - AutoModelForCausalLM, |
35 |
| - AutoModelForMaskedLM, |
36 |
| - AutoModelForMultipleChoice, |
37 |
| - AutoModelForQuestionAnswering, |
38 |
| - AutoModelForSeq2SeqLM, |
39 |
| - AutoModelForSequenceClassification, |
40 |
| - AutoModelForTokenClassification, |
41 |
| - AutoModelForVision2Seq, |
42 | 33 | DataCollator,
|
43 | 34 | PretrainedConfig,
|
44 | 35 | PreTrainedModel,
|
45 |
| - XLNetLMHeadModel, |
46 | 36 | default_data_collator,
|
47 | 37 | )
|
48 | 38 |
|
|
71 | 61 | INCModelForSequenceClassification,
|
72 | 62 | INCModelForTokenClassification,
|
73 | 63 | INCModelForVision2Seq,
|
74 |
| - INCModelForXLNetLM, |
75 | 64 | )
|
76 | 65 | from .utils import INCDataLoader, _cfgs_to_fx_cfgs
|
77 | 66 |
|
@@ -538,49 +527,3 @@ def _apply_quantization_from_config(q_config: Dict, model: torch.nn.Module) -> t
|
538 | 527 | q_model = convert(q_model, mapping=q_mapping, inplace=True)
|
539 | 528 |
|
540 | 529 | return q_model
|
541 |
| - |
542 |
| - |
543 |
| -class IncQuantizedModel(INCModel): |
544 |
| - @classmethod |
545 |
| - def from_pretrained(cls, *args, **kwargs): |
546 |
| - warnings.warn( |
547 |
| - f"The class `{cls.__name__}` has been depreciated and will be removed in optimum-intel v1.12, please use " |
548 |
| - f"`{cls.__name__.replace('IncQuantized', 'INC')}` instead." |
549 |
| - ) |
550 |
| - return super().from_pretrained(*args, **kwargs) |
551 |
| - |
552 |
| - |
553 |
| -class IncQuantizedModelForQuestionAnswering(IncQuantizedModel): |
554 |
| - auto_model_class = AutoModelForQuestionAnswering |
555 |
| - |
556 |
| - |
557 |
| -class IncQuantizedModelForSequenceClassification(IncQuantizedModel): |
558 |
| - auto_model_class = AutoModelForSequenceClassification |
559 |
| - |
560 |
| - |
561 |
| -class IncQuantizedModelForTokenClassification(IncQuantizedModel): |
562 |
| - auto_model_class = AutoModelForTokenClassification |
563 |
| - |
564 |
| - |
565 |
| -class IncQuantizedModelForMultipleChoice(IncQuantizedModel): |
566 |
| - auto_model_class = AutoModelForMultipleChoice |
567 |
| - |
568 |
| - |
569 |
| -class IncQuantizedModelForSeq2SeqLM(IncQuantizedModel): |
570 |
| - auto_model_class = AutoModelForSeq2SeqLM |
571 |
| - |
572 |
| - |
573 |
| -class IncQuantizedModelForCausalLM(IncQuantizedModel): |
574 |
| - auto_model_class = AutoModelForCausalLM |
575 |
| - |
576 |
| - |
577 |
| -class IncQuantizedModelForMaskedLM(IncQuantizedModel): |
578 |
| - auto_model_class = AutoModelForMaskedLM |
579 |
| - |
580 |
| - |
581 |
| -class IncQuantizedModelForXLNetLM(IncQuantizedModel): |
582 |
| - auto_model_class = XLNetLMHeadModel |
583 |
| - |
584 |
| - |
585 |
| -class IncQuantizedModelForVision2Seq(IncQuantizedModel): |
586 |
| - auto_model_class = AutoModelForVision2Seq |
0 commit comments