Spaces:
Runtime error
Runtime error
ronald
commited on
Commit
·
7999fec
1
Parent(s):
51e65cc
init
Browse files- 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 |
-
|
134 |
-
|
135 |
-
|
136 |
-
|
137 |
-
|
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":
|