Skip to content

Commit 894334d

Browse files
echarlaixecharlaix
authored
Fix format (#615)
1 parent 877cf9d commit 894334d

File tree

3 files changed

+70
-73
lines changed

3 files changed

+70
-73
lines changed

‎Makefile

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,11 @@ REAL_CLONE_URL = $(if $(CLONE_URL),$(CLONE_URL),$(DEFAULT_CLONE_URL))
2222
# Run code quality checks
2323
style_check:
2424
black --check .
25-
ruff .
25+
ruff check .
2626

2727
style:
2828
black .
29-
ruff . --fix
29+
ruff check . --fix
3030

3131
# Run tests for the library
3232
test:

‎examples/openvino/test_examples.py

+66-65
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,39 +30,39 @@
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

3839
class TestExamples(unittest.TestCase):
3940
def test_audio_classification(self):
4041
with tempfile.TemporaryDirectory() as tmp_dir:
4142
test_args = f"""
42-
run_audio_classification.py
43-
--model_name_or_path hf-internal-testing/tiny-random-Wav2Vec2Model
44-
--nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json
45-
--dataset_name superb
46-
--dataset_config_name ks
47-
--max_train_samples 10
48-
--max_eval_samples 2
49-
--remove_unused_columns False
50-
--do_train
51-
--learning_rate 3e-5
52-
--max_length_seconds 1
53-
--attention_mask False
54-
--warmup_ratio 0.1
55-
--num_train_epochs 1
56-
--gradient_accumulation_steps 1
57-
--dataloader_num_workers 1
58-
--logging_strategy steps
59-
--logging_steps 1
60-
--evaluation_strategy epoch
61-
--save_strategy epoch
62-
--load_best_model_at_end False
43+
run_audio_classification.py
44+
--model_name_or_path hf-internal-testing/tiny-random-Wav2Vec2Model
45+
--nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json
46+
--dataset_name superb
47+
--dataset_config_name ks
48+
--max_train_samples 10
49+
--max_eval_samples 2
50+
--remove_unused_columns False
51+
--do_train
52+
--learning_rate 3e-5
53+
--max_length_seconds 1
54+
--attention_mask False
55+
--warmup_ratio 0.1
56+
--num_train_epochs 1
57+
--gradient_accumulation_steps 1
58+
--dataloader_num_workers 1
59+
--logging_strategy steps
60+
--logging_steps 1
61+
--evaluation_strategy epoch
62+
--save_strategy epoch
63+
--load_best_model_at_end False
6364
--seed 42
64-
--output_dir {tmp_dir}
65+
--output_dir {tmp_dir}
6566
--overwrite_output_dir
6667
""".split()
6768

@@ -71,21 +72,21 @@ def test_audio_classification(self):
7172
def test_image_classification(self):
7273
with tempfile.TemporaryDirectory() as tmp_dir:
7374
test_args = f"""
74-
run_image_classification.py
75-
--model_name_or_path nateraw/vit-base-beans
76-
--dataset_name beans
77-
--max_train_samples 10
78-
--max_eval_samples 2
79-
--remove_unused_columns False
75+
run_image_classification.py
76+
--model_name_or_path nateraw/vit-base-beans
77+
--dataset_name beans
78+
--max_train_samples 10
79+
--max_eval_samples 2
80+
--remove_unused_columns False
8081
--do_train
81-
--do_eval
82-
--learning_rate 2e-5
83-
--num_train_epochs 1
84-
--logging_strategy steps
85-
--logging_steps 1
86-
--evaluation_strategy epoch
87-
--save_strategy epoch
88-
--save_total_limit 1
82+
--do_eval
83+
--learning_rate 2e-5
84+
--num_train_epochs 1
85+
--logging_strategy steps
86+
--logging_steps 1
87+
--evaluation_strategy epoch
88+
--save_strategy epoch
89+
--save_total_limit 1
8990
--seed 1337
9091
--output_dir {tmp_dir}
9192
""".split()
@@ -95,23 +96,23 @@ def test_image_classification(self):
9596

9697
def test_text_classification(self):
9798
with tempfile.TemporaryDirectory() as tmp_dir:
98-
test_args = f"""
99-
run_glue.py
100-
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForSequenceClassification
101-
--task_name sst2
102-
--max_train_samples 10
103-
--max_eval_samples 2
104-
--overwrite_output_dir
99+
test_args = f"""
100+
run_glue.py
101+
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForSequenceClassification
102+
--task_name sst2
103+
--max_train_samples 10
104+
--max_eval_samples 2
105+
--overwrite_output_dir
105106
--do_train
106-
--do_eval
107-
--max_seq_length 128
108-
--learning_rate 1e-5
109-
--optim adamw_torch
110-
--num_train_epochs 1
111-
--logging_steps 1
112-
--evaluation_strategy steps
113-
--eval_steps 1
114-
--save_strategy epoch
107+
--do_eval
108+
--max_seq_length 128
109+
--learning_rate 1e-5
110+
--optim adamw_torch
111+
--num_train_epochs 1
112+
--logging_steps 1
113+
--evaluation_strategy steps
114+
--eval_steps 1
115+
--save_strategy epoch
115116
--seed 42
116117
--output_dir {tmp_dir}
117118
""".split()
@@ -122,17 +123,17 @@ def test_text_classification(self):
122123
def test_question_answering(self):
123124
with tempfile.TemporaryDirectory() as tmp_dir:
124125
test_args = f"""
125-
run_qa.py
126-
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForQuestionAnswering
127-
--dataset_name squad
128-
--do_train
129-
--do_eval
130-
--max_train_samples 10
131-
--max_eval_samples 2
132-
--learning_rate 3e-5
133-
--num_train_epochs 1
134-
--max_seq_length 384
135-
--doc_stride 128
126+
run_qa.py
127+
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForQuestionAnswering
128+
--dataset_name squad
129+
--do_train
130+
--do_eval
131+
--max_train_samples 10
132+
--max_eval_samples 2
133+
--learning_rate 3e-5
134+
--num_train_epochs 1
135+
--max_seq_length 384
136+
--doc_stride 128
136137
--overwrite_output_dir
137138
--output_dir {tmp_dir}
138139
""".split()
@@ -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)