sedataskan commited on
Commit
1120639
·
verified ·
1 Parent(s): 8178b0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import torch
2
  import transformers
 
3
  from datasets import load_dataset
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments
5
  from peft import prepare_model_for_kbit_training, LoraConfig, get_peft_model, PeftModel
 
1
  import torch
2
  import transformers
3
+ import bitsandbytes
4
  from datasets import load_dataset
5
  from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments
6
  from peft import prepare_model_for_kbit_training, LoraConfig, get_peft_model, PeftModel