Kevin Fink commited on
Commit
41f77cb
·
1 Parent(s): 146a9de
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -267,7 +267,7 @@ def run_train(dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
267
  lora_dropout=0.1, # Dropout for LoRA layers
268
  bias="none" # Bias handling
269
  )
270
- model = get_peft_model(model, lora_config)
271
  result = fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad)
272
  return result
273
  # Create Gradio interface
 
267
  lora_dropout=0.1, # Dropout for LoRA layers
268
  bias="none" # Bias handling
269
  )
270
+ #model = get_peft_model(model, lora_config)
271
  result = fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad)
272
  return result
273
  # Create Gradio interface