yonikremer commited on
Commit
acb3322
1 Parent(s): 16ec708

added a small model (gpt 2) for debugging

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. 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
  )