Fix format

huggingface · Mar 19, 2024 · 55c481a · 55c481a
1 parent 877cf9d
commit 55c481a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git 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()
+    unittest.main()
diff --git 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()"
    ]