Spaces:
Sleeping
Sleeping
Commit ·
e7168ce
1
Parent(s): 113e3f4
update
Browse files
examples/tutorials/lora_unsloth/requirements.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
datasets
|
| 2 |
+
|
examples/tutorials/lora_unsloth/step_2_train_model.py
CHANGED
|
@@ -11,10 +11,10 @@ else:
|
|
| 11 |
project_path = os.path.abspath("../../../")
|
| 12 |
project_path = Path(project_path)
|
| 13 |
|
| 14 |
-
|
| 15 |
-
|
| 16 |
from datasets import load_dataset
|
| 17 |
-
|
| 18 |
|
| 19 |
|
| 20 |
def get_args():
|
|
|
|
| 11 |
project_path = os.path.abspath("../../../")
|
| 12 |
project_path = Path(project_path)
|
| 13 |
|
| 14 |
+
from unsloth import FastLanguageModel
|
| 15 |
+
from trl import SFTTrainer, SFTConfig
|
| 16 |
from datasets import load_dataset
|
| 17 |
+
import torch
|
| 18 |
|
| 19 |
|
| 20 |
def get_args():
|