diff --git a/examples/openvino/test_examples.py b/examples/openvino/test_examples.py index ac6f5afe9b..cad663d127 100644 --- a/examples/openvino/test_examples.py +++ b/examples/openvino/test_examples.py @@ -17,6 +17,7 @@ import unittest from unittest.mock import patch + SRC_DIRS = [ os.path.join(os.path.dirname(__file__), dirname) for dirname in [ @@ -29,9 +30,9 @@ sys.path.extend(SRC_DIRS) if SRC_DIRS is not None: - import run_image_classification import run_audio_classification import run_glue + import run_image_classification import run_qa @@ -142,4 +143,4 @@ def test_question_answering(self): if __name__ == "__main__": - unittest.main() \ No newline at end of file + unittest.main() diff --git a/notebooks/openvino/stable_diffusion_optimization.ipynb b/notebooks/openvino/stable_diffusion_optimization.ipynb index b44c00f78c..6c79bc5df0 100644 --- a/notebooks/openvino/stable_diffusion_optimization.ipynb +++ b/notebooks/openvino/stable_diffusion_optimization.ipynb @@ -69,9 +69,7 @@ "metadata": {}, "outputs": [], "source": [ - "quantized_pipe = OVStableDiffusionPipeline.from_pretrained(\n", - " \"OpenVINO/Stable-Diffusion-Pokemon-en-quantized\", compile=False\n", - ")\n", + "quantized_pipe = OVStableDiffusionPipeline.from_pretrained(\"OpenVINO/Stable-Diffusion-Pokemon-en-quantized\", compile=False)\n", "quantized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)\n", "quantized_pipe.compile()" ] @@ -104,9 +102,7 @@ "metadata": {}, "outputs": [], "source": [ - "optimized_pipe = OVStableDiffusionPipeline.from_pretrained(\n", - " \"OpenVINO/stable-diffusion-pokemons-tome-quantized\", compile=False\n", - ")\n", + "optimized_pipe = OVStableDiffusionPipeline.from_pretrained(\"OpenVINO/stable-diffusion-pokemons-tome-quantized\", compile=False)\n", "optimized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)\n", "optimized_pipe.compile()" ]