Spaces:
Runtime error
Runtime error
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:
|