mrcuddle commited on
Commit
c4a29da
1 Parent(s): fae0f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import logging
5
  from pathlib import Path
6
  import spaces
7
 
8
- @spaces.GPU(duration=300)
9
  def merge_and_upload(base_model, model_to_merge, scaling_factor, weight_drop_prob, repo_name, token, commit_message):
10
  # Define a fixed output path
11
  outpath = Path('/tmp')
 
5
  from pathlib import Path
6
  import spaces
7
 
8
+ @spaces.GPU()
9
  def merge_and_upload(base_model, model_to_merge, scaling_factor, weight_drop_prob, repo_name, token, commit_message):
10
  # Define a fixed output path
11
  outpath = Path('/tmp')