Upload app.py

app.py

@@ -12,7 +12,7 @@ import os
 from huggingface_hub import login
 from peft import get_peft_model, LoraConfig
 
-#os.environ['HF_HOME'] = '/data/.huggingface'
+os.environ['HF_HOME'] = '/data/.huggingface'
 
 @spaces.GPU(duration=120)
 def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
@@ -36,8 +36,8 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
     
         # Load the model and tokenizer
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2, force_download=True)
-        model.gradient_checkpointing_enable()
-        #model = get_peft_model(model, lora_config)
+        model = get_peft_model(model, lora_config)
+        model.gradient_checkpointing_enable()        
         
     
         # Set training arguments