Update precious3_gpt_multi_modal.py
Browse files
precious3_gpt_multi_modal.py
CHANGED
@@ -25,8 +25,12 @@ log = logging.getLogger(__name__)
|
|
25 |
|
26 |
|
27 |
class CustomTokenizer(PreTrainedTokenizerFast):
|
28 |
-
def __init__(self,
|
29 |
-
super().__init__(
|
|
|
|
|
|
|
|
|
30 |
|
31 |
|
32 |
class Custom_MptModel(MPTModel): # MptModel
|
|
|
25 |
|
26 |
|
27 |
class CustomTokenizer(PreTrainedTokenizerFast):
|
28 |
+
def __init__(self, **kwargs):
|
29 |
+
super().__init__( tokenizer_file="../tokenizer.json",
|
30 |
+
unk_token="[UNK]",
|
31 |
+
pad_token="[PAD]",
|
32 |
+
eos_token="[EOS]",
|
33 |
+
bos_token="[BOS]", **kwargs)
|
34 |
|
35 |
|
36 |
class Custom_MptModel(MPTModel): # MptModel
|