Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
eaidova committed Aug 27, 2024
1 parent d4f4cc1 commit 789a7ac
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -447,6 +447,8 @@
"metadata": {},
"outputs": [],
"source": [
"core = ov.Core()\n",
"\n",
"class OVMobileLlamaForCausalLM(transformers.GenerationMixin):\n",
" def __init__(self, stage1_path, stage2_path, device):\n",
" self.stage1 = core.compile_model(stage1_path, device)\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -235,7 +235,6 @@
"\n",
"\n",
"example_input = {\n",
"import openvino as ov\n",
" \"input_ids\": torch.ones((9, 1), dtype=torch.int64),\n",
" \"encoder_hidden_states\": torch.ones((1, 39, 1024), dtype=torch.float32),\n",
" \"encoder_attention_mask\": torch.ones((1, 39), dtype=torch.int64),\n",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@
"source": [
"import platform\n",
"\n",
"%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu \"gradio>=4.19\" \"openvino>=2023.3.0\" \"transformers>=4.37\" \"torch>=2.1\" Pillow sentencepiece protobuf scipy datasets nncf\n",
"%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu \"gradio>=4.19\" \"protobuf>=3.20.3\" \"openvino>=2023.3.0\" \"transformers>=4.37\" \"torch>=2.1\" Pillow sentencepiece protobuf scipy datasets nncf\n",
"\n",
"if platform.system() != \"Windows\":\n",
" %pip install -q \"matplotlib>=3.4\"\n",
Expand Down

0 comments on commit 789a7ac

Please sign in to comment.