silveroxides commited on
Commit
f9d4d6e
·
1 Parent(s): 3099eb9

fix: Remove torch.cuda.is_available() to prevent CUDA init in main process

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -149,8 +149,7 @@ def upload_model_as_pr(
149
  @spaces.GPU(duration=30)
150
  def completion_signal():
151
  """Brief GPU allocation at completion to satisfy ZeroGPU requirements."""
152
- import torch
153
- return torch.cuda.is_available()
154
 
155
 
156
  def run_quantization(config):
 
149
  @spaces.GPU(duration=30)
150
  def completion_signal():
151
  """Brief GPU allocation at completion to satisfy ZeroGPU requirements."""
152
+ return True
 
153
 
154
 
155
  def run_quantization(config):