John6666 commited on
Commit
bb9b216
β€’
1 Parent(s): 970c031

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def change_base_model(repo_id: str, cn_on: bool): # , progress=gr.Progress(track
73
  print(f"Loading model: {repo_id}")
74
  #clear_cache()
75
  pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=dtype, vae=taef1).to(device)
76
- pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
77
  last_model = repo_id
78
  last_cn_on = cn_on
79
  #progress(1, desc=f"Model loaded: {repo_id}")
 
73
  print(f"Loading model: {repo_id}")
74
  #clear_cache()
75
  pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=dtype, vae=taef1).to(device)
76
+ #pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
77
  last_model = repo_id
78
  last_cn_on = cn_on
79
  #progress(1, desc=f"Model loaded: {repo_id}")