apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): do not parse tokens when removing them

This commit is contained in:
Sean Sube 2023-11-14 20:34:56 -06:00
parent 14ddfb43bb
commit 1bdfb4ac8a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,7 +3,7 @@ from copy import deepcopy
from logging import getLogger from logging import getLogger
from math import ceil from math import ceil
from re import Pattern, compile from re import Pattern, compile
from typing import Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
import numpy as np import numpy as np
import torch import torch
@ -515,9 +515,13 @@ def parse_reseed(prompt: str) -> Tuple[str, List[Reseed]]:
return get_tokens_from_prompt(prompt, RESEED_TOKEN, parser=parse_reseed_group) return get_tokens_from_prompt(prompt, RESEED_TOKEN, parser=parse_reseed_group)
def skip_group(group) -> Any:
return group
def remove_tokens(prompt: Optional[str]) -> Optional[str]: def remove_tokens(prompt: Optional[str]) -> Optional[str]:
if prompt is None: if prompt is None:
return prompt return prompt
remainder, tokens = get_tokens_from_prompt(prompt, ANY_TOKEN) remainder, tokens = get_tokens_from_prompt(prompt, ANY_TOKEN, parser=skip_group)
return remainder return remainder