Spaces:
Running
Running
Update my_model/fine_tuner/fine_tuner.py
Browse files
my_model/fine_tuner/fine_tuner.py
CHANGED
@@ -27,7 +27,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TrainingArguments,
|
|
27 |
from trl import SFTTrainer
|
28 |
from datasets import Dataset, load_dataset
|
29 |
from peft import LoraConfig, prepare_model_for_kbit_training, PeftModel
|
30 |
-
import fine_tuning_config as config
|
31 |
from typing import List
|
32 |
import bitsandbytes # only on GPU
|
33 |
import gc
|
|
|
27 |
from trl import SFTTrainer
|
28 |
from datasets import Dataset, load_dataset
|
29 |
from peft import LoraConfig, prepare_model_for_kbit_training, PeftModel
|
30 |
+
import my_model.config.fine_tuning_config as config
|
31 |
from typing import List
|
32 |
import bitsandbytes # only on GPU
|
33 |
import gc
|