Check for gptqmodel first when exporting a GPTQ model to openvino

Author: notsyncing

optimum/exporters/openvino/__main__.py

@@ -39,6 +39,7 @@
     _infer_library_from_model_name_or_path,
     _OpenClipForZeroShotImageClassification,
 )
+from optimum.utils import is_auto_gptq_available, is_gptqmodel_available
 from optimum.utils.save_utils import maybe_load_preprocessors
 
 from .utils import (
@@ -339,7 +340,12 @@ def main_export(
                 orig_post_init_model = GPTQQuantizer.post_init_model
 
                 def post_init_model(self, model):
-                    from auto_gptq import exllama_set_max_input_length
+                    if is_gptqmodel_available():
+                        from gptqmodel import exllama_set_max_input_length
+                    elif is_auto_gptq_available():
+                        from auto_gptq import exllama_set_max_input_length
+                    else:
+                        raise ValueError("neither auto_gptq nor gptqmodel is available")
 
                     class StoreAttr(object):
                         pass