coh mech

ccl_win.py

@@ -129,7 +129,7 @@ class ccl_win(evaluate.Measurement):
             device = "cuda" if torch.cuda.is_available() else "cpu"
 
         tokenizer = AutoTokenizer.from_pretrained("roberta-large")
-        model = transformers.AutoModelForSequenceClassification.from_pretrained(f"./{dataset}/", num_labels=2,cache_dir=MODEL_CACHE_DIR)
+        model = AutoModelForSequenceClassification.from_pretrained(f"./{dataset}/", num_labels=2,cache_dir=MODEL_CACHE_DIR)
         model.to(device)
 
         pred_list,len_by_sample = self.preprocess_adjacent_window(preds)