crystal-technologies commited on
Commit
e3e15c4
·
1 Parent(s): 96d3447

Update Perceptrix/engine.py

Browse files
Files changed (1) hide show
  1. 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)