yonikremer
commited on
Commit
•
acb3322
1
Parent(s):
16ec708
added a small model (gpt 2) for debugging
Browse files- README.md +1 -1
- available_models.py +2 -1
README.md
CHANGED
@@ -8,7 +8,7 @@ sdk_version: 1.17.0
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
fullWidth: true
|
11 |
-
models: [decapoda-research/llama-30b-hf, decapoda-research/llama-13b-hf, decapoda-research/llama-7b-hf]
|
12 |
tags: [text-generation, pytorch, transformers, streamlit]
|
13 |
---
|
14 |
|
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
fullWidth: true
|
11 |
+
models: [decapoda-research/llama-30b-hf, decapoda-research/llama-13b-hf, decapoda-research/llama-7b-hf, gpt2]
|
12 |
tags: [text-generation, pytorch, transformers, streamlit]
|
13 |
---
|
14 |
|
available_models.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
AVAILABLE_MODELS = (
|
2 |
"decapoda-research/llama-30b-hf",
|
3 |
"decapoda-research/llama-13b-hf",
|
4 |
-
"decapoda-research/llama-7b-hf"
|
|
|
5 |
)
|
|
|
1 |
AVAILABLE_MODELS = (
|
2 |
"decapoda-research/llama-30b-hf",
|
3 |
"decapoda-research/llama-13b-hf",
|
4 |
+
"decapoda-research/llama-7b-hf",
|
5 |
+
"gpt2",
|
6 |
)
|