Spaces:
Running
on
Zero
Running
on
Zero
wondervictor
commited on
Commit
·
7d33904
1
Parent(s):
60990f1
add requirements
Browse files- language/t5.py +1 -1
language/t5.py
CHANGED
@@ -38,7 +38,7 @@ class T5Embedder:
|
|
38 |
'config.json', 'special_tokens_map.json', 'spiece.model', 'tokenizer_config.json',
|
39 |
'pytorch_model.bin.index.json', 'pytorch_model-00001-of-00002.bin', 'pytorch_model-00002-of-00002.bin'
|
40 |
]:
|
41 |
-
hf_hub_download(repo_id=f'
|
42 |
force_filename=filename, token=self.hf_token)
|
43 |
tokenizer_path, path = cache_dir, cache_dir
|
44 |
else:
|
|
|
38 |
'config.json', 'special_tokens_map.json', 'spiece.model', 'tokenizer_config.json',
|
39 |
'pytorch_model.bin.index.json', 'pytorch_model-00001-of-00002.bin', 'pytorch_model-00002-of-00002.bin'
|
40 |
]:
|
41 |
+
hf_hub_download(repo_id=f'google/{dir_or_name}', filename=filename, cache_dir=cache_dir,
|
42 |
force_filename=filename, token=self.hf_token)
|
43 |
tokenizer_path, path = cache_dir, cache_dir
|
44 |
else:
|