init

app.py

@@ -21,7 +21,7 @@ class LoggingCallback(TrainerCallback):
 
 @spaces.GPU
 def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
-    try:
+    try:        
         login(api_key.strip())
         lora_config = LoraConfig(
             r=16,  # Rank of the low-rank adaptation
@@ -62,7 +62,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             logging_steps=10,
             #push_to_hub=True,
             hub_model_id=hub_id.strip(),
-            #fp16=True,
+            fp16=True,
             #lr_scheduler_type='cosine',
         )