silveroxides commited on
Commit
5c45880
1 Parent(s): c8eac63

Update convert.py

Browse files
Files changed (1) hide show
  1. convert.py +1 -1
convert.py CHANGED
@@ -25,7 +25,7 @@ if torch.cuda.is_available():
25
  else:
26
  torch_dtype = torch.float32
27
 
28
- @spaces.GPU(duration=5)
29
  def convert_single(model_id: str, filename: str, folder: str, progress: Any, token: str):
30
  progress(0, desc="Downloading model")
31
  local_file = os.path.join(model_id, filename)
 
25
  else:
26
  torch_dtype = torch.float32
27
 
28
+ @spaces.GPU()
29
  def convert_single(model_id: str, filename: str, folder: str, progress: Any, token: str):
30
  progress(0, desc="Downloading model")
31
  local_file = os.path.join(model_id, filename)