Fix notebooks (#603)

- 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)
huggingface · Mar 15, 2024 · cd3bc8a · cd3bc8a
1 parent 948f99d
commit cd3bc8a
Show file tree

Hide file tree

Showing 3 changed files with 228 additions and 212 deletions.
diff --git a/.github/workflows/test_openvino_notebooks.yml b/.github/workflows/test_openvino_notebooks.yml
@@ -49,5 +49,7 @@ jobs:
 
     - name: Test with Pytest
       run: |
+        sed -i 's/NUM_TRAIN_ITEMS = 600/NUM_TRAIN_ITEMS = 10/' notebooks/openvino/question_answering_quantization.ipynb
+        sed -i 's/# %pip install/%pip install/' notebooks/openvino/optimum_openvino_inference.ipynb
         python -m pytest --nbval-lax notebooks/openvino/optimum_openvino_inference.ipynb  notebooks/openvino/question_answering_quantization.ipynb