Remove warning

661047f1 · Casper · GitHub · 1d0e1285 · 661047f1
Unverified Commit 661047f1 authored Oct 05, 2023 by Casper Committed by GitHub Oct 05, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 10 deletions

awq/models/mistral.py awq/models/mistral.py +2 -10

No files found.
--- a/awq/models/mistral.py
+++ b/awq/models/mistral.py
-import logging
 from typing import Dict
 from .base import BaseAWQForCausalLM
-
-try:
-    from transformers.models.mistral.modeling_mistral import MistralDecoderLayer, MistralForCausalLM
-except:
-    # TODO: Remove once released on PyPi
-    logging.warning("You need the latest transformers 4.34.0.dev0: pip install -U git+https://github.com/huggingface/transformers.git")
-    MistralForCausalLM = None
-    MistralDecoderLayer = None
+from transformers.models.mistral.modeling_mistral import MistralDecoderLayer, MistralForCausalLM

 class MistralAWQForCausalLM(BaseAWQForCausalLM):
    layer_type = "MistralDecoderLayer"