Kevin Fink commited on
Commit
b529f79
·
1 Parent(s): 6fc7b98
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from transformers import Trainer, TrainingArguments, AutoTokenizer, AutoModelFor
4
  from transformers import DataCollatorForSeq2Seq
5
  from datasets import load_dataset
6
  import traceback
 
7
  from huggingface_hub import login
8
  from peft import get_peft_model, LoraConfig
9
 
 
4
  from transformers import DataCollatorForSeq2Seq
5
  from datasets import load_dataset
6
  import traceback
7
+ import os
8
  from huggingface_hub import login
9
  from peft import get_peft_model, LoraConfig
10