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

fix: Add @spaces.GPU with extended duration for ZeroGPU compatibility

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,