Lora commited on
Commit
71ffc8f
1 Parent(s): abc2084

change model to stanfordnlp and force gradio version to be exactly 3.32.0

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
 
7
  # Build model & get some layers
8
  tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
9
- m = AutoModelForCausalLM.from_pretrained("lora-x/backpack-gpt2", trust_remote_code=True)
10
  m.eval()
11
 
12
  lm_head = m.get_lm_head() # (V, d)
 
6
 
7
  # Build model & get some layers
8
  tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
9
+ m = AutoModelForCausalLM.from_pretrained("stanfordnlp/backpack-gpt2", trust_remote_code=True)
10
  m.eval()
11
 
12
  lm_head = m.get_lm_head() # (V, d)
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  torch
2
  pandas
3
  transformers
4
- gradio>=3.32.0
 
1
  torch
2
  pandas
3
  transformers
4
+ gradio==3.32.0