diff --git a/gpt.py b/gpt.py index 267d669..30cf9d6 100644 --- a/gpt.py +++ b/gpt.py @@ -61,7 +61,7 @@ def main(): examples["text"], truncation=True, padding="max_length", - max_length=2048, # Zwiększono dla dłuższych artykułów + max_length=1024, # Zwiększono dla dłuższych artykułów return_tensors="pt" ) tokenized["labels"] = tokenized["input_ids"].clone()