Upload app.py

app.py

@@ -41,7 +41,7 @@ controlnet = None
 last_model = models[0]
 last_cn_on = False
 controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype).to(device)
-controlnet = FluxMultiControlNetModel([controlnet_union])#.to(device)
+controlnet = FluxMultiControlNetModel([controlnet_union]).to(device)
 
 
 MAX_SEED = 2**32-1
@@ -102,7 +102,7 @@ def preload_base_model(repo_id: str, cn_on: bool, progress=gr.Progress(track_tqd
             clear_cache()
             #controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)
             #controlnet = FluxMultiControlNetModel([controlnet_union])
-            pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype)
+            pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=None, torch_dtype=dtype)
             print(f"Model Chached: {repo_id} / ControlNet Cahced: {controlnet_model_union_repo}")
         else:
             print(f"Caching Model: {repo_id}")