ronaldahmed commited on
Commit
65798f3
1 Parent(s): 2fefeaf

update cache dir server

Browse files
Files changed (1) hide show
  1. ccl_win.py +3 -1
ccl_win.py CHANGED
@@ -118,11 +118,13 @@ class ccl_win(evaluate.Measurement):
118
  def _compute(self, predictions, dataset="arxiv", batch_size: int = 16, device=None, use_aggregator=True):
119
  """Returns the scores"""
120
  MODEL_CACHE_DIR = "/home/rcardena/.cache/huggingface/"
121
- BASEDIR = "/gfs/team/nlp/users/rcardena/tools/new_evals/ccl_win"
122
  if getpass.getuser() == "s1987051":
123
  MODEL_CACHE_DIR="/disk/ocean/rcardenas/tools/huggingface/"
124
  elif getpass.getuser() == "rcardena":
125
  MODEL_CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/"
 
 
126
 
127
  if device is not None:
128
  # assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
 
118
  def _compute(self, predictions, dataset="arxiv", batch_size: int = 16, device=None, use_aggregator=True):
119
  """Returns the scores"""
120
  MODEL_CACHE_DIR = "/home/rcardena/.cache/huggingface/"
121
+ BASEDIR = "/bask/projects/j/jlxi8926-auto-sum/rcardenas/tools/ccl_win"
122
  if getpass.getuser() == "s1987051":
123
  MODEL_CACHE_DIR="/disk/ocean/rcardenas/tools/huggingface/"
124
  elif getpass.getuser() == "rcardena":
125
  MODEL_CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/"
126
+ elif getpass.getuser() == "gvhr8913":
127
+ MODEL_CACHE_DIR="/bask/projects/j/jlxi8926-auto-sum/rcardenas/cache"
128
 
129
  if device is not None:
130
  # assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."