m7mdal7aj commited on
Commit
0c4272d
1 Parent(s): b739d2a

Update my_model/fine_tuner/fine_tuner.py

Browse files
Files changed (1) hide show
  1. my_model/fine_tuner/fine_tuner.py +1 -1
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