Skip to content

Commit

Permalink
Chasing bug
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelbenayoun committed Jul 19, 2024
1 parent e309fde commit 71640d6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/cli/test_neuron_cache_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def test_optimum_neuron_cache_add(self, hub_test):
with open(tmpdir / "bert_vocab.txt", "w") as fp:
fp.write("\n".join([get_random_string(random.randint(10, 20))] + mandatory_tokens))

tokenizer = BertTokenizer(tmpdir / "bert_vocab.txt")
tokenizer = BertTokenizer((tmpdir / "bert_vocab.txt").as_posix())
tokenizer.save_pretrained(bert_model_name)

model = BertModel(config)
Expand All @@ -118,7 +118,7 @@ def test_optimum_neuron_cache_add(self, hub_test):
with open(tmpdir / "t5_vocab.txt", "w") as fp:
fp.write("\n".join([get_random_string(random.randint(10, 20))] + mandatory_tokens))

tokenizer = T5Tokenizer(tmpdir / "t5_vocab.txt")
tokenizer = T5Tokenizer((tmpdir / "t5_vocab.txt").as_posix())
tokenizer.save_pretrained(t5_model_name)

model = T5Model(config)
Expand Down

0 comments on commit 71640d6

Please sign in to comment.