jill/LLMsForDisinformationAnalysis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to CPU only, as to not confuse GPU

Browse Source
This commit is contained in:
William Jeynes
2026-03-31 16:09:41 +01:00
parent 36c30a427d
commit 893829e599
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
supporting/RAGAS_Service/ensemble_service.py
+1 -1
View File
@@ -92,7 +92,7 @@ LABEL_TO_INT = {v: k for k, v in INT_TO_LABEL.items()}
flan_tokenizer = AutoTokenizer.from_pretrained(FLAN_PATH) flan_tokenizer = AutoTokenizer.from_pretrained(FLAN_PATH)
flan_model = AutoModelForSeq2SeqLM.from_pretrained(FLAN_PATH) flan_model = AutoModelForSeq2SeqLM.from_pretrained(FLAN_PATH)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cpu")
flan_model.to(device) flan_model.to(device)
flan_model.eval() flan_model.eval()