up

app.py

@@ -5,6 +5,7 @@ from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 from optimization import get_compiled_transformer
 from hub_utils import _push_compiled_graph_to_hub
 from huggingface_hub import whoami
+import spaces
 
 # --- Model Loading ---
 dtype = torch.bfloat16
@@ -17,6 +18,7 @@ pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 # --- Ahead-of-time compilation ---
 compiled_transformer = get_compiled_transformer(pipe, prompt="prompt")
 
+@spaces.GPU(duration=120)
 def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken):
     if not filename.endswith(".pt2"):
         raise NotImplementedError("The filename must end with a `.pt2` extension.")