Fixed mypy issues

Author: suryasidd

examples/openvino/stable_diffusion/export_lcm.py

@@ -4,7 +4,7 @@
 # except in compliance with the License. See the license file found in the
 # LICENSE file in the root directory of this source tree.
 
-# mypy: disable-error-code=union-attr
+# mypy: disable-error-code="union-attr,import-not-found"
 
 import argparse
 import logging
@@ -15,7 +15,7 @@
 from torch.export import export
 
 try:
-    from diffusers import DiffusionPipeline
+    from diffusers import DiffusionPipeline  # type: ignore[import-not-found]
 except ImportError:
     raise ImportError(
         "Please install diffusers and transformers: pip install diffusers transformers"

examples/openvino/stable_diffusion/openvino_lcm.py

@@ -4,7 +4,7 @@
 # except in compliance with the License. See the license file found in the
 # LICENSE file in the root directory of this source tree.
 
-# mypy: disable-error-code=union-attr
+# mypy: disable-error-code="union-attr,import-not-found"
 
 import argparse
 import logging