change hyper parameters

modules/llm/Mistral-7b/Mistral-7b_fine_tune.py

@@ -58,7 +58,6 @@ def fine_tune(base_model, new_model):
     training_arguments = TrainingArguments(
         per_device_train_batch_size=2,
         gradient_accumulation_steps=1,
-        num_train_epochs=2,
         learning_rate=1e-4,
         logging_steps=2,
         optim="adamw_torch",

modules/llm/Mixtral-8x7b/Mixtral-8x7b_fine_tune.py

@@ -72,7 +72,6 @@ def fine_tuned(base_model, new_model):
         args=TrainingArguments(
             per_device_train_batch_size=2,
             gradient_accumulation_steps=1,
-            num_train_epochs=2,
             learning_rate=1e-4,
             logging_steps=2,
             optim="adamw_torch",