Kevin Fink
commited on
Commit
·
f9e951b
1
Parent(s):
038610e
init
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from peft import get_peft_model, LoraConfig
|
|
9 |
@spaces.GPU
|
10 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|
11 |
try:
|
12 |
-
login(api_key)
|
13 |
lora_config = LoraConfig(
|
14 |
r=16, # Rank of the low-rank adaptation
|
15 |
lora_alpha=32, # Scaling factor
|
|
|
9 |
@spaces.GPU
|
10 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|
11 |
try:
|
12 |
+
login(api_key.strip())
|
13 |
lora_config = LoraConfig(
|
14 |
r=16, # Rank of the low-rank adaptation
|
15 |
lora_alpha=32, # Scaling factor
|