Hugging Face's logo

Spaces:
K00B404
/
Merged_Models_Test_Chat
Runtime error

App Files Community
K00B404 commited on
Commit
a32decb
·
verified ·
1 Parent(s): 0d9d5f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
  import torch
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
11
- #model_id = 'EleutherAI/pythia-1b'
12
- model_id = 'EleutherAI/GPT-Neo-2.7B'
13
  #gr.load(f"models/{model_id}").launch()
14
  tokenizer = AutoTokenizer.from_pretrained(model_id)
15
 
 
8
  import torch
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
11
+ #model_id = 'EleutherAI/pythia-1b' # 16
12
+ model_id = 'EleutherAI/GPT-Neo-2.7B' #32 layers
13
  #gr.load(f"models/{model_id}").launch()
14
  tokenizer = AutoTokenizer.from_pretrained(model_id)
15