up

app.py

@@ -18,7 +18,7 @@ pipe.transformer.fuse_qkv_projections()
 pipe.transformer.set_attn_processor(FlashFusedFluxAttnProcessor3_0())
 
 @spaces.GPU(duration=1200)
-def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken, progress=gr.Progress(track_tqdm=True)):
+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.")
 

optimization.py

@@ -41,5 +41,7 @@ def compile_transformer(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.k
         print("Export done.")
         return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
 
+    print(f"{pipeline.transformer.device=}")
     compiled_transformer = f()
+    print("Compilation done.")
     return compiled_transformer