Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Apr 5, 2024

Commit

269f149

1 Parent(s): f8debfa

Corrected typo

Files changed (1) hide show

scripts.py CHANGED Viewed

@@ -122,7 +122,7 @@ def train_models(path,progress=gradio.Progress(track_tqdm=True)):
     tokenizer = tokenizers.Tokenizer.from_pretrained('roberta-base')
     trainer = qarac.models.QaracTrainerModel.QaracTrainerModel('roberta-base',
                                                                tokenizer)
-    device = torch.device('cude:0')
     trainer.to(device)
     loss_fn = CombinedLoss()
     loss_fn.cuda()

     tokenizer = tokenizers.Tokenizer.from_pretrained('roberta-base')
     trainer = qarac.models.QaracTrainerModel.QaracTrainerModel('roberta-base',
                                                                tokenizer)
+    device = torch.device('cuda:0')
     trainer.to(device)
     loss_fn = CombinedLoss()
     loss_fn.cuda()