init

app.py

@@ -19,7 +19,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
     
         # Load the model and tokenizer
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name, num_labels=2)
-        model = get_peft_model(model, lora_config)
+        #model = get_peft_model(model, lora_config)
         tokenizer = AutoTokenizer.from_pretrained(model_name)
     
         # Tokenize the dataset
@@ -48,7 +48,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
             push_to_hub=True,
             hub_model_id=hub_id,
             fp16=True,
-            lr_scheduler_type='cosine',
+            #lr_scheduler_type='cosine',
         )
         
         # Create Trainer