root commited on
Commit
8251ef9
1 Parent(s): 5b3c5ad
modeling_mpt.py CHANGED
@@ -23,6 +23,7 @@ Tokenizer = Union[PreTrainedTokenizer, PreTrainedTokenizerFast]
23
  class MPTPreTrainedModel(PreTrainedModel):
24
  config_class = MPTConfig
25
  base_model_prefix = 'model'
 
26
 
27
  class MPTModel(MPTPreTrainedModel):
28
 
@@ -287,4 +288,4 @@ class MPTForCausalLM(MPTPreTrainedModel):
287
  reordered_past = []
288
  for layer_past in past_key_values:
289
  reordered_past += [tuple((past_state.index_select(0, beam_idx) for past_state in layer_past))]
290
- return reordered_past
 
23
  class MPTPreTrainedModel(PreTrainedModel):
24
  config_class = MPTConfig
25
  base_model_prefix = 'model'
26
+ trust_remote_code= True
27
 
28
  class MPTModel(MPTPreTrainedModel):
29
 
 
288
  reordered_past = []
289
  for layer_past in past_key_values:
290
  reordered_past += [tuple((past_state.index_select(0, beam_idx) for past_state in layer_past))]
291
+ return reordered_past
pytorch_model-00001-of-00002.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5782a8714b23c6f85c9433411df36de8c2ffac0008b5fd4df20f78fe592990f
3
+ size 9943040275
pytorch_model-00002-of-00002.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89642468caaca82ffe684b4c98f3f53249c180d6c99f5895f5be9afeea656f98
3
+ size 3355599187