John6666 commited on
Commit
766e9b2
·
verified ·
1 Parent(s): 04314b2

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def change_base_model(repo_id: str, cn_on: bool): # , progress=gr.Progress(track
66
  #controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)#.to(device)
67
  #controlnet = FluxMultiControlNetModel([controlnet_union])#.to(device)
68
  #pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype)#.to(device)
69
- pipe = FluxControlNetPipeline.from_pretrained(repo_id, torch_dtype=dtype)
70
  #pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
71
  last_model = repo_id
72
  last_cn_on = cn_on
 
66
  #controlnet_union = FluxControlNetModel.from_pretrained(controlnet_model_union_repo, torch_dtype=dtype)#.to(device)
67
  #controlnet = FluxMultiControlNetModel([controlnet_union])#.to(device)
68
  #pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=controlnet, torch_dtype=dtype)#.to(device)
69
+ pipe = FluxControlNetPipeline.from_pretrained(repo_id, controlnet=None, torch_dtype=dtype)
70
  #pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
71
  last_model = repo_id
72
  last_cn_on = cn_on