Spaces:

vector2000
/

test_finetunning

Paused

vector2000 commited on Aug 27, 2024

Commit

2cdb9f1

verified ·

1 Parent(s): 2b42ea0

Update finetunning.py

Files changed (1) hide show

finetunning.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 import time
 import torch
-from datasets import load_dataset # Dataset
 from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,

 import os
 import time
 import torch
+from datasets import Dataset
 from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,