fix model variable name

huggingface · Apr 30, 2024 · a4e142b · a4e142b
1 parent 39954b1
commit a4e142b
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/notebooks/setfit-optimum-intel.ipynb b/notebooks/setfit-optimum-intel.ipynb
@@ -903,9 +903,6 @@
     "setfit_model.model_body.tokenizer = model.tokenizer\n",
     "setfit_model.model_body.device = model.device\n",
     "\n",
-    "body = torch.jit.trace(body, (d,), check_trace=False, strict=False)\n",
-    "setfit_model.model_body = torch.jit.freeze(body)\n",
-    "            \n",
     "pb = PerformanceBenchmark(\n",
     "    model=setfit_model,\n",
     "    dataset=test_dataset,\n",