Upload app.py

app.py

@@ -90,8 +90,6 @@ def load_quantized_control(control_repo: str, dtype, hf_token):
     return transformer, text_encoder_2
 
 def load_pipeline(repo_id: str, cn_on: bool, model_type: str, task: str, dtype_str: str, hf_token: str, progress=gr.Progress(track_tqdm=True)):
-    pipe = None
-    pipe_i2i = None
     try:
         controlnet_model_union_repo = 'InstantX/FLUX.1-dev-Controlnet-Union'
         if task == "Flux Fill" or repo_id in models_fill:
@@ -169,8 +167,8 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, mode
         taef1.to("cpu")
         if controlnet is not None: controlnet.to("cpu")
         if controlnet_union is not None: controlnet_union.to("cpu")
+        pipe, pipe_i2i = load_pipeline(pipe, pipe_i2i, repo_id, cn_on, model_type, task, dtype_str, HF_TOKEN, progress)
         clear_cache()
-        pipe, pipe_i2i = load_pipeline(repo_id, cn_on, model_type, task, dtype_str, HF_TOKEN, progress)
         last_model = repo_id
         last_cn_on = cn_on
         last_task = task