Skip to content

Commit 55c481a

Browse files
committed
Fix format
1 parent 877cf9d commit 55c481a

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

examples/openvino/test_examples.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
import unittest
1818
from unittest.mock import patch
1919

20+
2021
SRC_DIRS = [
2122
os.path.join(os.path.dirname(__file__), dirname)
2223
for dirname in [
@@ -29,9 +30,9 @@
2930
sys.path.extend(SRC_DIRS)
3031

3132
if SRC_DIRS is not None:
32-
import run_image_classification
3333
import run_audio_classification
3434
import run_glue
35+
import run_image_classification
3536
import run_qa
3637

3738

@@ -142,4 +143,4 @@ def test_question_answering(self):
142143

143144

144145
if __name__ == "__main__":
145-
unittest.main()
146+
unittest.main()

notebooks/openvino/stable_diffusion_optimization.ipynb

+2-6
Original file line numberDiff line numberDiff line change
@@ -69,9 +69,7 @@
6969
"metadata": {},
7070
"outputs": [],
7171
"source": [
72-
"quantized_pipe = OVStableDiffusionPipeline.from_pretrained(\n",
73-
" \"OpenVINO/Stable-Diffusion-Pokemon-en-quantized\", compile=False\n",
74-
")\n",
72+
"quantized_pipe = OVStableDiffusionPipeline.from_pretrained(\"OpenVINO/Stable-Diffusion-Pokemon-en-quantized\", compile=False)\n",
7573
"quantized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)\n",
7674
"quantized_pipe.compile()"
7775
]
@@ -104,9 +102,7 @@
104102
"metadata": {},
105103
"outputs": [],
106104
"source": [
107-
"optimized_pipe = OVStableDiffusionPipeline.from_pretrained(\n",
108-
" \"OpenVINO/stable-diffusion-pokemons-tome-quantized\", compile=False\n",
109-
")\n",
105+
"optimized_pipe = OVStableDiffusionPipeline.from_pretrained(\"OpenVINO/stable-diffusion-pokemons-tome-quantized\", compile=False)\n",
110106
"optimized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)\n",
111107
"optimized_pipe.compile()"
112108
]

0 commit comments

Comments
 (0)