EveSa commited on
Commit
23e12c6
1 Parent(s): 52ab81f

add cache dir to inference_t5

Browse files
Files changed (1) hide show
  1. src/inference_t5.py +2 -0
src/inference_t5.py CHANGED
@@ -3,6 +3,8 @@
3
  """
4
  import re
5
  import string
 
 
6
 
7
  import contractions
8
  import torch
 
3
  """
4
  import re
5
  import string
6
+ import os
7
+ os.environ['TRANSFORMERS_CACHE'] = './.cache'
8
 
9
  import contractions
10
  import torch