Spaces:
Build error
Build error
Commit ·
c22291c
1
Parent(s): 4127859
fix: Add @spaces.GPU with extended duration for ZeroGPU compatibility
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ FP8/INT8 formats for ComfyUI inference, with HuggingFace Hub integration.
|
|
| 7 |
import os
|
| 8 |
import tempfile
|
| 9 |
import gradio as gr
|
|
|
|
| 10 |
from huggingface_hub import hf_hub_download, HfApi, create_commit, CommitOperationAdd
|
| 11 |
|
| 12 |
from convert_to_quant import convert, ConversionConfig
|
|
@@ -146,6 +147,7 @@ def upload_model_as_pr(
|
|
| 146 |
except Exception as e:
|
| 147 |
return f"❌ Upload failed: {str(e)}"
|
| 148 |
|
|
|
|
| 149 |
def quantize_model(
|
| 150 |
source_repo: str,
|
| 151 |
file_path: str,
|
|
|
|
| 7 |
import os
|
| 8 |
import tempfile
|
| 9 |
import gradio as gr
|
| 10 |
+
import spaces # ZeroGPU
|
| 11 |
from huggingface_hub import hf_hub_download, HfApi, create_commit, CommitOperationAdd
|
| 12 |
|
| 13 |
from convert_to_quant import convert, ConversionConfig
|
|
|
|
| 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,
|