apply review comments

Author: eaidova

optimum/exporters/openvino/__main__.py

@@ -410,10 +410,10 @@ class StoreAttr(object):
                     )
                 model.config.pad_token_id = pad_token_id
 
-        if hasattr(model.config, "export_model_type") and model.config.export_model_type is not None:
+        if hasattr(model.config, "export_model_type"):
             model_type = model.config.export_model_type.replace("_", "-")
         else:
-            model_type = (getattr(model.config, "model_type", "") or "").replace("_", "-")
+            model_type = model.config.model_type.replace("_", "-")
 
         if (
             not custom_architecture

optimum/exporters/openvino/convert.py

@@ -18,7 +18,6 @@
 import inspect
 import logging
 import os
-import types
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Union
 
@@ -1115,10 +1114,7 @@ def get_ltx_video_models_for_export(pipeline, exporter, int_dtype, float_dtype):
         }
     )
 
-    def _vae_decode_forward(self, latent_sample, timestep=None):
-        return self.decode(z=latent_sample, temb=timestep)
-
-    vae_decoder.forward = types.MethodType(_vae_decode_forward, vae_decoder)
+    vae_decoder.forward = lambda latent_sample, timestep=None: vae_decoder.decode(z=latent_sample, temb=timestep)
 
     vae_config_constructor = TasksManager.get_exporter_config_constructor(
         model=vae_decoder,

tests/openvino/test_diffusion.py

@@ -1006,7 +1006,7 @@ def test_textual_inversion(self):
         np.testing.assert_allclose(ov_output, diffusers_output, atol=1e-4, rtol=1e-2)
 
 
-@unittest.skipIf(is_transformers_version("<", "4.45"))
+@unittest.skipIf(is_transformers_version("<", "4.45"), "Required transformers >= 4.45")
 class OVPipelineForText2VideoTest(unittest.TestCase):
     SUPPORTED_ARCHITECTURES = []
     if is_transformers_version(">=", "4.45.0"):