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():
|
def main():
|
||||||
# Inicjalizacja tokenizera
|
# 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.add_special_tokens({"additional_special_tokens": SPECIAL_TOKENS})
|
||||||
tokenizer.pad_token = tokenizer.eos_token
|
tokenizer.pad_token = tokenizer.eos_token
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue