Skip to content
Snippets Groups Projects
Commit 27967a29 authored by Julien Breton's avatar Julien Breton
Browse files

add missing imports

parent 8333bad9
No related branches found
No related tags found
No related merge requests found
......@@ -3,7 +3,7 @@ import transformers
from datasets import load_dataset
from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments
from peft import prepare_model_for_kbit_training, LoraConfig, get_peft_model, PeftModel
import transformers, json
import transformers, json, importlib
from alive_progress import alive_bar
import bitsandbytes, flash_attn
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment