From ec248f7dbc4ca0325120fce38d1bb577c49403dd Mon Sep 17 00:00:00 2001 From: Alexander Date: Fri, 28 Jun 2024 11:12:05 +0400 Subject: [PATCH] Fixed another failing notebook --- notebooks/openvino/optimum_openvino_inference.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/notebooks/openvino/optimum_openvino_inference.ipynb b/notebooks/openvino/optimum_openvino_inference.ipynb index dcd7dc866f..76c77aec55 100644 --- a/notebooks/openvino/optimum_openvino_inference.ipynb +++ b/notebooks/openvino/optimum_openvino_inference.ipynb @@ -539,7 +539,7 @@ "ov_pipe = pipeline(\"audio-classification\", model=model, feature_extractor=feature_extractor)\n", "\n", "# streaming=true enables loading one item from the dataset without downloading the full dataset\n", - "dataset = load_dataset(\"speech_commands\", \"v0.02\", streaming=True)\n", + "dataset = load_dataset(\"speech_commands\", \"v0.02\", streaming=True, trust_remote_code=True)\n", "audio_sample = next(iter(dataset[\"test\"]))" ] }, @@ -894,8 +894,8 @@ "from transformers import AutoTokenizer, pipeline\n", "\n", "model_id = \"helenai/t5-small-ov\"\n", - "model = OVModelForSeq2SeqLM.from_pretrained(model_id, compile=False)\n", - "tokenizer = AutoTokenizer.from_pretrained(model_id)\n", + "model = OVModelForSeq2SeqLM.from_pretrained(model_id, compile=False, trust_remote_code=True)\n", + "tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)\n", "\n", "input_ids = tokenizer(\"translate English to German: This house is wonderful.\", return_tensors=\"pt\").input_ids\n", "outputs = model.generate(input_ids, max_new_tokens=10)\n",