John6666 commited on
Commit
ecc15b8
·
verified ·
1 Parent(s): cb87640

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1892,10 +1892,10 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1892
 
1893
  def generate_minimal(
1894
  args: List[Any],
1895
- model_name: str,
1896
- vae_model: str,
1897
- task: str,
1898
- controlnet_model: str
1899
  ) -> Tuple[str, Optional[List[str]], Optional[str]]:
1900
  # Preload the requested model (drain loader yields so the model is ready)
1901
  for _ in sd_gen.load_new_model(model_name, vae_model, task, controlnet_model):
 
1892
 
1893
  def generate_minimal(
1894
  args: List[Any],
1895
+ model_name: str=LOAD_DIFFUSERS_FORMAT_MODEL[0],
1896
+ vae_model: Optional[str]=None,
1897
+ task: str=TASK_MODEL_LIST[0],
1898
+ controlnet_model: str=DIFFUSERS_CONTROLNET_MODEL[0]
1899
  ) -> Tuple[str, Optional[List[str]], Optional[str]]:
1900
  # Preload the requested model (drain loader yields so the model is ready)
1901
  for _ in sd_gen.load_new_model(model_name, vae_model, task, controlnet_model):