dev

app.py

@@ -126,7 +126,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
                     args=training_args,
                     train_dataset=train_dataset,
                     eval_dataset=saved_test_dataset,
-                    #compute_metrics=compute_metrics,
+                    compute_metrics=compute_metrics,
                 )
             
             elif os.access(f'/data/{hub_id.strip()}_train_dataset3', os.R_OK):                        
@@ -227,7 +227,7 @@ def run_train(dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
         lora_dropout=0.1,  # Dropout for LoRA layers
         bias="none"  # Bias handling
     )
-    model = get_peft_model(model, lora_config)
+    #model = get_peft_model(model, lora_config)
     result = fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad)
     return result
 # Create Gradio interface