Kevin Fink
commited on
Commit
·
1e083b4
1
Parent(s):
3ae598a
dev
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ import os
|
|
| 12 |
from huggingface_hub import login
|
| 13 |
from peft import get_peft_model, LoraConfig
|
| 14 |
|
| 15 |
-
os.environ['HF_HOME'] = '/data/.huggingface'
|
| 16 |
|
| 17 |
@spaces.GPU(duration=120)
|
| 18 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|
|
|
|
| 12 |
from huggingface_hub import login
|
| 13 |
from peft import get_peft_model, LoraConfig
|
| 14 |
|
| 15 |
+
#os.environ['HF_HOME'] = '/data/.huggingface'
|
| 16 |
|
| 17 |
@spaces.GPU(duration=120)
|
| 18 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|