Spaces:

nroggendorff
/

train-llama

Runtime error

nroggendorff commited on Jul 17

Commit

f970cca

•

1 Parent(s): 315ce9c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -114,7 +114,7 @@ trainer = trl.SFTTrainer(
     args=args,
     train_dataset=dataset,
     dataset_text_field='text',
-    max_seq_length=512,
 )
 torch.cuda.set_device(0)

     args=args,
     train_dataset=dataset,
     dataset_text_field='text',
+    max_seq_length=512
 )
 torch.cuda.set_device(0)