Upload app.py

app.py

@@ -48,7 +48,7 @@ pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_ret
 # https://huggingface.co/InstantX/FLUX.1-dev-Controlnet-Union
 # https://huggingface.co/spaces/jiuface/FLUX.1-dev-Controlnet-Union
 @spaces.GPU()
-def change_base_model(repo_id: str, cn_on: bool):
+def change_base_model(repo_id: str, cn_on: bool, progress=gr.Progress(track_tqdm=True)):
     global pipe
     global controlnet_union
     global controlnet
@@ -59,7 +59,7 @@ def change_base_model(repo_id: str, cn_on: bool):
         if cn_on:
             #progress(0, desc=f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
             print(f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
-            clear_cache()
+            #clear_cache()
             controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype).to(device)
             controlnet = FluxMultiControlNetModel([controlnet_union]).to(device)
             pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype).to(device)
@@ -71,7 +71,7 @@ def change_base_model(repo_id: str, cn_on: bool):
         else:
             #progress(0, desc=f"Loading model: {repo_id}")
             print(f"Loading model: {repo_id}")
-            clear_cache()
+            #clear_cache()
             pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=dtype, vae=taef1).to(device)
             pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
             last_model = repo_id