Spaces:

ronaldahmed
/

ccl_win

Sleeping

ronald commited on Apr 26, 2023

Commit

092069b

1 Parent(s): 0e62420

coh mech

Files changed (1) hide show

ccl_win.py CHANGED Viewed

@@ -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.to(device)
         pred_list,len_by_sample = self.preprocess_adjacent_window(preds)

             device = "cuda" if torch.cuda.is_available() else "cpu"
         tokenizer = AutoTokenizer.from_pretrained("roberta-large")
+        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)