update

huggingface · Jan 12, 2024 · 8966e11 · 8966e11
1 parent 4d25ed2
commit 8966e11
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
@@ -1682,9 +1682,11 @@ def standardize_model_attributes(
             if "Transformer" in model[0].__class__.__name__:
                 model.config = model[0].auto_model.config
                 model.config.model_type = "sentence-transformers-transformer"
+                model.config.__class__.model_type = "sentence-transformers-transformer"
             elif "CLIP" in model[0].__class__.__name__:
                 model.config = model[0].model.config
                 model.config.model_type = "sentence-transformers-clip"
+                model.config.__class__.model_type = "sentence-transformers-clip"
             else:
                 raise ValueError(
                     f"The export of a sentence-transformers model with the first module being {model[0].__class__.__name__} is currently not supported in Optimum. Please open an issue or submit a PR to add the support."