Spaces:
Running
on
Zero
Running
on
Zero
Upload app.py
Browse files
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):
|