lora-x commited on
Commit
5f69038
1 Parent(s): ffb38f8

Use the backpack model under stanfordnlp

Browse files
Files changed (1) hide show
  1. app.py +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)