up

app.py

@@ -12,8 +12,11 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the model pipeline
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/Flux.1-Dev", torch_dtype=dtype).to(device)
+pipe = DiffusionPipeline.from_pretrained(
+    "black-forest-labs/Flux.1-Dev", torch_dtype=dtype
+).to(device)
 
+@spaces.GPU(duration=1500)
 def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken, progress=gr.Progress(track_tqdm=True)):
     if not filename.endswith(".pt2"):
         raise NotImplementedError("The filename must end with a `.pt2` extension.")

optimization.py

@@ -27,7 +27,6 @@ INDUCTOR_CONFIGS = {
 
 
 def compile_transformer(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
-    @spaces.GPU(duration=1500)
     def f():
         with spaces.aoti_capture(pipeline.transformer) as call:
             pipeline(*args, **kwargs)