Commit
·
e3e15c4
1
Parent(s):
96d3447
Update Perceptrix/engine.py
Browse files- Perceptrix/engine.py +1 -1
Perceptrix/engine.py
CHANGED
@@ -5,7 +5,7 @@ import transformers
|
|
5 |
import torch
|
6 |
import tqdm
|
7 |
|
8 |
-
model_name = "mosaicml/mpt-30b"
|
9 |
|
10 |
config = transformers.AutoConfig.from_pretrained(
|
11 |
model_name, trust_remote_code=True)
|
|
|
5 |
import torch
|
6 |
import tqdm
|
7 |
|
8 |
+
model_name = "mosaicml/mpt-30b-instruct"
|
9 |
|
10 |
config = transformers.AutoConfig.from_pretrained(
|
11 |
model_name, trust_remote_code=True)
|