apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

parse region coordinates

This commit is contained in:
Sean Sube 2023-11-05 16:38:43 -06:00
parent b365660adc
commit ee2173405a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/diffusers/utils.py
View File

@ -458,5 +458,10 @@ def slice_prompt(prompt: str, slice: int) -> str:
Region = Tuple[int, int, int, int, Literal["add", "replace"], str] Region = Tuple[int, int, int, int, Literal["add", "replace"], str]
def parse_region_group(group) -> Region:
top, left, bottom, right, mode, prompt = group
return (int(top), int(left), int(bottom), int(right), mode, prompt)
def parse_regions(prompt: str) -> Tuple[str, List[Region]]: def parse_regions(prompt: str) -> Tuple[str, List[Region]]:
return get_tokens_from_prompt(prompt, REGION_TOKEN, lambda it: it) return get_tokens_from_prompt(prompt, REGION_TOKEN, parser=parse_region_group)