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- app.py +1 -1
- 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("
|
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
|
|
|
1 |
torch
|
2 |
pandas
|
3 |
transformers
|
4 |
+
gradio==3.32.0
|