Update app.py
Browse files
app.py
CHANGED
@@ -47,14 +47,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
47 |
# model_id = "mistralai/Mistral-7B-v0.3"
|
48 |
|
49 |
model_id = "01-ai/Yi-1.5-9B-Chat"
|
50 |
-
|
51 |
|
52 |
|
53 |
tokenizer = AutoTokenizer.from_pretrained(
|
54 |
# model_id
|
55 |
model_id,
|
56 |
# use_fast=False
|
57 |
-
|
58 |
|
59 |
|
60 |
accelerator = Accelerator()
|
|
|
47 |
# model_id = "mistralai/Mistral-7B-v0.3"
|
48 |
|
49 |
model_id = "01-ai/Yi-1.5-9B-Chat"
|
50 |
+
model_id = "Qwen/Qwen2-7B-Instruct"
|
51 |
|
52 |
|
53 |
tokenizer = AutoTokenizer.from_pretrained(
|
54 |
# model_id
|
55 |
model_id,
|
56 |
# use_fast=False
|
57 |
+
token= token,)
|
58 |
|
59 |
|
60 |
accelerator = Accelerator()
|