silveroxides commited on
Commit
aee2946
·
1 Parent(s): c22291c

fix: Move GPU decorator to minimal stub function, call at end of quantization

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -147,7 +147,13 @@ def upload_model_as_pr(
147
  except Exception as e:
148
  return f"❌ Upload failed: {str(e)}"
149
 
150
- @spaces.GPU(duration=600) # Extended duration for large models
 
 
 
 
 
 
151
  def quantize_model(
152
  source_repo: str,
153
  file_path: str,
@@ -248,6 +254,9 @@ def quantize_model(
248
  )
249
  status_log.append(upload_status)
250
 
 
 
 
251
  return result.output_path, "\n\n".join(status_log)
252
 
253
  except Exception as e:
 
147
  except Exception as e:
148
  return f"❌ Upload failed: {str(e)}"
149
 
150
+ @spaces.GPU(duration=30)
151
+ def gpu_check():
152
+ """Minimal GPU function to satisfy ZeroGPU space requirements."""
153
+ import torch
154
+ return torch.cuda.is_available()
155
+
156
+
157
  def quantize_model(
158
  source_repo: str,
159
  file_path: str,
 
254
  )
255
  status_log.append(upload_status)
256
 
257
+ # Brief GPU check to satisfy ZeroGPU requirements
258
+ gpu_check()
259
+
260
  return result.output_path, "\n\n".join(status_log)
261
 
262
  except Exception as e: