Addressing comments

Signed-off-by: Asmita Goswami <[email protected]>

Author: asmigosw

QEfficient/base/common.py

@@ -14,10 +14,10 @@
 
 from typing import Any
 
-from QEfficient.transformers.modeling_utils import model_class_mapping
 from transformers import AutoConfig
 
 from QEfficient.base.modeling_qeff import QEFFBaseModel
+from QEfficient.transformers.modeling_utils import model_class_mapping
 
 
 class QEFFCommonLoader:

QEfficient/cloud/infer.py

@@ -16,7 +16,7 @@
 from transformers.models.auto.modeling_auto import MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES
 
 from QEfficient.base.common import QEFFCommonLoader
-from QEfficient.utils import check_and_assign_cache_dir, load_hf_tokenizer, constants
+from QEfficient.utils import check_and_assign_cache_dir, constants, load_hf_tokenizer
 from QEfficient.utils.logging_utils import logger
 
 

QEfficient/transformers/modeling_utils.py

@@ -8,10 +8,9 @@
 from collections import namedtuple
 from typing import Dict, Optional, Tuple, Type
 
-import transformers.models.auto.modeling_auto as mapping
-
 import torch
 import torch.nn as nn
+import transformers.models.auto.modeling_auto as mapping
 from transformers.models.codegen.modeling_codegen import (
     CodeGenAttention,
     CodeGenBlock,