Update train_model.py

train_model.py

@@ -100,7 +100,8 @@ def initialize_model(task, model_name, vocab_size, sequence_length, hidden_size,
                 num_attention_heads=attention_heads,
                 intermediate_size=4 * hidden_size,
                 hidden_act='gelu',
-                use_cache=True
+                use_cache=True,
+                truncation=False
             )
             model = GPT2LMHeadModel(config)
             logging.info("GPT2LMHeadModel initialized successfully.")