Skip to content

Commit 869a0ba

Browse files
committed
Fix notebooks
- use dataset for audio example to fix permission issue with URL - change pytorch_model.bin to model.safetensors - speed up quantization notebook test by using only 10 samples - update half() messaging (no longer needed to do half() on GPU)
1 parent 358f389 commit 869a0ba

File tree

3 files changed

+228
-212
lines changed

3 files changed

+228
-212
lines changed

.github/workflows/test_openvino_notebooks.yml

+2
Original file line numberDiff line numberDiff line change
@@ -49,5 +49,7 @@ jobs:
4949

5050
- name: Test with Pytest
5151
run: |
52+
sed -i 's/NUM_TRAIN_ITEMS = 600/NUM_TRAIN_ITEMS = 10/' notebooks/openvino/question_answering_quantization.ipynb
53+
sed -i 's/# %pip install/%pip install/' notebooks/openvino/optimum_openvino_inference.ipynb
5254
python -m pytest --nbval-lax notebooks/openvino/optimum_openvino_inference.ipynb notebooks/openvino/question_answering_quantization.ipynb
5355

0 commit comments

Comments
 (0)