Ali-C137 commited on
Commit
a912e42
1 Parent(s): 990aba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -5,11 +5,11 @@ import gradio as gr
5
 
6
  ### need more GPU power to call better models !!!!!!
7
 
8
- model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp", use_cache=True) # 11B param
9
- tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")
10
 
11
- #model = AutoModelForCausalLM.from_pretrained('EleutherAI/gpt-neo-1.3B', use_cache=True)
12
- #tokenizer = AutoTokenizer.from_pretrained('EleutherAI/gpt-neo-1.3B')
13
 
14
  set_seed(424242)
15
 
 
5
 
6
  ### need more GPU power to call better models !!!!!!
7
 
8
+ #model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp", use_cache=True) # 11B param
9
+ #tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")
10
 
11
+ model = AutoModelForCausalLM.from_pretrained('EleutherAI/gpt-neo-1.3B', use_cache=True)
12
+ tokenizer = AutoTokenizer.from_pretrained('EleutherAI/gpt-neo-1.3B')
13
 
14
  set_seed(424242)
15