apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correctly handle missing inversion param

This commit is contained in:
Sean Sube 2023-02-25 12:54:51 -06:00
parent 973b0a9352
commit 3626d69f40
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/params.py
View File

@ -157,7 +157,7 @@ class ImageParams:
lpw: bool = False, lpw: bool = False,
eta: float = 0.0, eta: float = 0.0,
batch: int = 1, batch: int = 1,
inversion: str = None, inversion: Optional[str] = None,
) -> None: ) -> None:
self.model = model self.model = model
self.scheduler = scheduler self.scheduler = scheduler

5
api/onnx_web/serve.py
View File

@ -159,7 +159,10 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
scheduler = get_from_map( scheduler = get_from_map(
request.args, "scheduler", pipeline_schedulers, get_config_value("scheduler") request.args, "scheduler", pipeline_schedulers, get_config_value("scheduler")
) )
inversion = get_not_empty(request.args, "inversion", get_config_value("inversion"))
inversion = request.args.get("inversion", None)
inversion_path = None
if inversion is not None and inversion.strip() != "":
inversion_path = get_model_path(inversion) inversion_path = get_model_path(inversion)
# image params # image params

6
common/scripts/copy-pr-to-gitlab.sh Normal file
View File

@ -0,0 +1,6 @@
PR_USER=$1
PR_BRANCH=$2
git remote add $1 git@github.com:$1/onnx-web.git
git fetch $1
git push gitlab refs/remotes/$1/$2:refs/heads/$1-$2