legacy107 commited on
Commit
a1fb3d5
1 Parent(s): 690e852

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
  model_name = "legacy107/flan-t5-large-bottleneck-adapter-cpgQA-unique"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name, device_map="auto")
7
  model = AutoModelForSeq2SeqLM.from_pretrained(
8
- model_checkpoint, device_map="auto"
9
  )
10
  model.set_active_adapters("question_answering")
11
 
 
5
  model_name = "legacy107/flan-t5-large-bottleneck-adapter-cpgQA-unique"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name, device_map="auto")
7
  model = AutoModelForSeq2SeqLM.from_pretrained(
8
+ model_name, device_map="auto"
9
  )
10
  model.set_active_adapters("question_answering")
11