Spaces:
Running
on
Zero
Running
on
Zero
Upload app.py
Browse files
app.py
CHANGED
|
@@ -2073,7 +2073,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
|
|
| 2073 |
|
| 2074 |
# 4) Implementations: both share the same signature via clone_signature
|
| 2075 |
@clone_signature(_signature_src)
|
| 2076 |
-
def generate_image(
|
| 2077 |
_load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
|
| 2078 |
argv = _argv_from_kwargs(kwargs)
|
| 2079 |
last = ("COMPLETE", None, None)
|
|
@@ -2082,7 +2082,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
|
|
| 2082 |
return last
|
| 2083 |
|
| 2084 |
@clone_signature(_signature_src)
|
| 2085 |
-
def generate_image_stream(
|
| 2086 |
_load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
|
| 2087 |
argv = _argv_from_kwargs(kwargs)
|
| 2088 |
yield from _generate_image(argv)
|
|
|
|
| 2073 |
|
| 2074 |
# 4) Implementations: both share the same signature via clone_signature
|
| 2075 |
@clone_signature(_signature_src)
|
| 2076 |
+
def generate_image(**kwargs) -> Tuple[str, Optional[List[str]], Optional[str]]:
|
| 2077 |
_load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
|
| 2078 |
argv = _argv_from_kwargs(kwargs)
|
| 2079 |
last = ("COMPLETE", None, None)
|
|
|
|
| 2082 |
return last
|
| 2083 |
|
| 2084 |
@clone_signature(_signature_src)
|
| 2085 |
+
def generate_image_stream(**kwargs) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
|
| 2086 |
_load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
|
| 2087 |
argv = _argv_from_kwargs(kwargs)
|
| 2088 |
yield from _generate_image(argv)
|