Upload app.py

app.py

@@ -89,7 +89,7 @@ def load_quantized_control(control_repo: str, dtype, hf_token):
     text_encoder_2 = T5EncoderModel.from_pretrained(control_repo, subfolder="text_encoder_2", torch_dtype=dtype, token=hf_token).to("cpu")
     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)):
+def load_pipeline(pipe, pipe_i2i, repo_id: str, cn_on: bool, model_type: str, task: str, dtype_str: str, hf_token: str, progress=gr.Progress(track_tqdm=True)):
     try:
         controlnet_model_union_repo = 'InstantX/FLUX.1-dev-Controlnet-Union'
         if task == "Flux Fill" or repo_id in models_fill: