Spaces:

lamekemal
/

brvm_finetuner

Sleeping

lamekemal commited on Sep 23

Commit

2f398e8

verified ·

1 Parent(s): 098b2be

Update finetune.py

Files changed (1) hide show

finetune.py CHANGED Viewed

@@ -4,7 +4,9 @@ from transformers import (
     AutoTokenizer,
     AutoModelForCausalLM,
     TrainingArguments,
 )
 from peft import LoraConfig, get_peft_model
 from trl import SFTTrainer
 import os

     AutoTokenizer,
     AutoModelForCausalLM,
     TrainingArguments,
+    BitsAndBytesConfig,
 )
 from peft import LoraConfig, get_peft_model
 from trl import SFTTrainer
 import os