fix UnboundLocalError

app_onediff.py

@@ -60,7 +60,6 @@ base_pipe = AutoPipelineForText2Image.from_pretrained(
 base_pipe.to(device)
 
 # base_pipe = nexfort_compile(base_pipe)
-compiled_pipe = base_pipe
 
 from gfpgan.utils import GFPGANer
 from basicsr.archs.srvgg_arch import SRVGGNetCompact
@@ -85,16 +84,14 @@ def create_demo() -> gr.Blocks:
         prompt:str,
         steps:int,
     ):
+        global compiled_pipe
         run_task_time = 0
         time_cost_str = ''
         run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-        if id(compiled_pipe) == id(base_pipe):
+        if not compiled_pipe:
             print("---------->Recompile<----------")
-            target_pipe = nexfort_compile(base_pipe)
-            compiled_pipe = target_pipe
-        else:
-            target_pipe = compiled_pipe
-        generated_image = target_pipe(
+            compiled_pipe = nexfort_compile(base_pipe)
+        generated_image = compiled_pipe(
             prompt=prompt,
             num_inference_steps=steps,
         ).images[0]