johnhew lora-x commited on
Commit
e1d9b31
1 Parent(s): 0dcac17

Use the backpack model under stanfordnlp (#2)

Browse files

- Use the backpack model under stanfordnlp (5f69038c5250417234deede3b01cfab8c712db27)


Co-authored-by: Lora Xie <lora-x@users.noreply.huggingface.co>

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)