Spaces:
Runtime error
Runtime error
Commit
·
da85b3a
1
Parent(s):
8c245db
test
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import time
|
|
| 4 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 5 |
|
| 6 |
# Load models and tokenizers
|
| 7 |
-
model_names = ["lmsys/vicuna-7b", "gpt2"]
|
| 8 |
models = [AutoModelForCausalLM.from_pretrained(name) for name in model_names]
|
| 9 |
tokenizers = [AutoTokenizer.from_pretrained(name) for name in model_names]
|
| 10 |
|
|
|
|
| 4 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 5 |
|
| 6 |
# Load models and tokenizers
|
| 7 |
+
model_names = ["lmsys/vicuna-7b-v1.3", "gpt2"]
|
| 8 |
models = [AutoModelForCausalLM.from_pretrained(name) for name in model_names]
|
| 9 |
tokenizers = [AutoTokenizer.from_pretrained(name) for name in model_names]
|
| 10 |
|