ronald commited on
Commit
7999fec
·
1 Parent(s): 51e65cc
Files changed (1) hide show
  1. local_coh_ppl.py +6 -6
local_coh_ppl.py CHANGED
@@ -130,12 +130,12 @@ class LocalCohPPL(evaluate.Measurement):
130
 
131
  ## PEDICTIONS: [str] sentences joined by "\n"
132
  def _compute(self, predictions, model_id, batch_size: int = 16, add_start_token: bool = True, device=None):
133
- MODEL_CACHE_DIR = "/home/rcardena/.cache/huggingface/"
134
- if getpass.getuser() == "s1987051":
135
- MODEL_CACHE_DIR="/disk/ocean/rcardenas/tools/huggingface/"
136
- elif getpass.getuser() == "rcardena"::
137
- MODEL_CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/"
138
-
139
  if device is not None:
140
  assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
141
  if device == "gpu":
 
130
 
131
  ## PEDICTIONS: [str] sentences joined by "\n"
132
  def _compute(self, predictions, model_id, batch_size: int = 16, add_start_token: bool = True, device=None):
133
+ MODEL_CACHE_DIR = "/home/rcardena/.cache/huggingface/"
134
+ if getpass.getuser() == "s1987051":
135
+ MODEL_CACHE_DIR="/disk/ocean/rcardenas/tools/huggingface/"
136
+ elif getpass.getuser() == "rcardena"::
137
+ MODEL_CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/"
138
+
139
  if device is not None:
140
  assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
141
  if device == "gpu":