From 6e537ee6fdc202361fb52495a6cc2f41a1cbe16f Mon Sep 17 00:00:00 2001 From: Toolkit User Date: Thu, 8 Jun 2023 14:12:44 +0000 Subject: [PATCH] Updated model name. --- embedding_sandbox.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/embedding_sandbox.ipynb b/embedding_sandbox.ipynb index c07152e..a8547bb 100644 --- a/embedding_sandbox.ipynb +++ b/embedding_sandbox.ipynb @@ -57,7 +57,7 @@ " super().__init__()\n", "\n", " self.model_name = model_name\n", - " self.tokenizer = prepare_tokenizer(model_name, use_auth_token=True)\n", + " self.tokenizer = prepare_tokenizer(model_name)\n", " self.encoder = AutoModel.from_pretrained(model_name, use_auth_token=True).to(DEVICE).eval()\n", " self.device = device\n", " self.max_input_len = max_input_len\n",