coh mech

ccl_win.py

@@ -128,7 +128,7 @@ class ccl_win(evaluate.Measurement):
         else:
             device = "cuda" if torch.cuda.is_available() else "cpu"
 
-        tokenizer = AutoTokenizer.from_pretrained("roberta-large",cache_dir=MODEL_CACHE_DIR,use_fast="cnn_dailymail" not in dataset)
+        tokenizer = AutoTokenizer.from_pretrained("roberta-large")
         model = transformers.AutoModelForSequenceClassification.from_pretrained(f"./{dataset}/", num_labels=2,cache_dir=MODEL_CACHE_DIR)
         model.to(device)