Kevin Fink
commited on
Commit
·
41f77cb
1
Parent(s):
146a9de
deve
Browse files
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
|