mod gpt
This commit is contained in:
parent
2dd8198c3a
commit
0df49895cf
5
gpt.py
5
gpt.py
|
|
@ -16,7 +16,10 @@ def prepare_simple_dataset():
|
|||
|
||||
def main():
|
||||
# Inicjalizacja tokenizera
|
||||
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
||||
tokenizer = AutoTokenizer.from_pretrained(
|
||||
MODEL_NAME,
|
||||
mean_resizing=False
|
||||
)
|
||||
tokenizer.add_special_tokens({"additional_special_tokens": SPECIAL_TOKENS})
|
||||
tokenizer.pad_token = tokenizer.eos_token
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue