update CTRL tokenizer pretrained path

guillaume-be · Sep 30, 2023 · 9b32309 · 9b32309
1 parent c632261
commit 9b32309
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main/Cargo.lock b/main/Cargo.lock
diff --git a/python-bindings/tests/test_tokenization_sst2.py b/python-bindings/tests/test_tokenization_sst2.py
@@ -123,7 +123,7 @@ def test_tokenization_distilbert(self):
 
     def test_tokenization_ctrl(self):
         # Given
-        self.base_tokenizer = CTRLTokenizer.from_pretrained('ctrl',
+        self.base_tokenizer = CTRLTokenizer.from_pretrained('Salesforce/ctrl',
                                                             do_lower_case=True,
                                                             cache_dir=self.test_dir)
         self.rust_tokenizer = PyCtrlTokenizer(