root
commited on
Commit
•
8251ef9
1
Parent(s):
5b3c5ad
add
Browse files- modeling_mpt.py +2 -1
- pytorch_model-00001-of-00002.bin +3 -0
- pytorch_model-00002-of-00002.bin +3 -0
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
|