Merged_Models_Test_Chat

Runtime error

K00B404 commited on Apr 15

Commit

f5127e3

•

1 Parent(s): 9dac1fa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_id = 'TinyPixel/Llama-2-7B-bf16-sharded'
 #gr.load(f"models/{model_id}").launch()
 tokenizer = AutoTokenizer.from_pretrained(model_id)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_id = 'EleutherAI/pythia-1b'
 #gr.load(f"models/{model_id}").launch()
 tokenizer = AutoTokenizer.from_pretrained(model_id)