Fix notebooks

- use local audio file to fix permission issue with datasets 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 12, 2024 · b1f2676 · b1f2676
1 parent 358f389
commit b1f2676
Show file tree

Hide file tree

Showing 4 changed files with 184 additions and 199 deletions.
diff --git a/.github/workflows/test_openvino_notebooks.yml b/.github/workflows/test_openvino_notebooks.yml
@@ -49,5 +49,6 @@ jobs:
 
     - name: Test with Pytest
       run: |
+        sed -i 's/NUM_TRAIN_ITEMS = 600/NUM_TRAIN_ITEMS = 10/' notebooks/openvino/question_answering_quantization.ipynb
         python -m pytest --nbval-lax notebooks/openvino/optimum_openvino_inference.ipynb  notebooks/openvino/question_answering_quantization.ipynb
 
diff --git a/notebooks/openvino/data/audio.mp3 b/notebooks/openvino/data/audio.mp3