wondervictor commited on
Commit
7d33904
·
1 Parent(s): 60990f1

add requirements

Browse files
Files changed (1) hide show
  1. 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'DeepFloyd/{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:
 
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: