Kevin Fink commited on
Commit
bab0771
·
1 Parent(s): 0a86c5b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  from huggingface_hub import login
9
  from peft import get_peft_model, LoraConfig
10
 
11
- os.environ('HF_HOME', '/data/.huggingface')
12
 
13
  @spaces.GPU(duration=120)
14
  def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
 
8
  from huggingface_hub import login
9
  from peft import get_peft_model, LoraConfig
10
 
11
+ os.environ['HF_HOME'] = '/data/.huggingface'
12
 
13
  @spaces.GPU(duration=120)
14
  def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):