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
|