apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-07-04 12:09:46 -05:00
parent 6ec7777f77
commit d1ecef9f2e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/base.py
View File

@ -1,7 +1,7 @@
from datetime import timedelta from datetime import timedelta
from logging import getLogger from logging import getLogger
from time import monotonic from time import monotonic
from typing import Any, List, Optional, Protocol, Tuple from typing import Any, List, Optional, Tuple
from PIL import Image from PIL import Image

1
api/onnx_web/params.py
View File

@ -186,7 +186,6 @@ class ImageParams:
steps: int steps: int
seed: int seed: int
negative_prompt: Optional[str] negative_prompt: Optional[str]
lpw: bool
eta: float eta: float
batch: int batch: int
control: Optional[NetworkModel] control: Optional[NetworkModel]

7
api/onnx_web/utils.py
View File

@ -6,7 +6,7 @@ from json import JSONDecodeError
from logging import getLogger from logging import getLogger
from os import environ, path from os import environ, path
from platform import system from platform import system
from typing import Any, Dict, List, Optional, Sequence, Union, TypeVar from typing import Any, Dict, List, Optional, Sequence, TypeVar, Union
import torch import torch
from yaml import safe_load from yaml import safe_load
@ -45,6 +45,7 @@ def get_and_clamp_int(
TElem = TypeVar("TElem") TElem = TypeVar("TElem")
def get_from_list( def get_from_list(
args: Any, key: str, values: Sequence[TElem], default_value: Optional[TElem] = None args: Any, key: str, values: Sequence[TElem], default_value: Optional[TElem] = None
) -> Optional[TElem]: ) -> Optional[TElem]:
@ -59,7 +60,9 @@ def get_from_list(
return None return None
def get_from_map(args: Any, key: str, values: Dict[str, TElem], default: TElem) -> TElem: def get_from_map(
args: Any, key: str, values: Dict[str, TElem], default: TElem
) -> TElem:
selected = args.get(key, default) selected = args.get(key, default)
if selected in values: if selected in values:
return values[selected] return values[selected]