Spaces:
Build error
Build error
Commit
·
f9d4d6e
1
Parent(s):
3099eb9
fix: Remove torch.cuda.is_available() to prevent CUDA init in main process
Browse files
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 |
-
|
| 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):
|