1
0
Fork 0

lint(api): include more params in JSON response and file

This commit is contained in:
Sean Sube 2023-02-05 10:50:15 -06:00
parent 0a9f108156
commit 4a728cb9ae
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 18 additions and 7 deletions

View File

@ -192,7 +192,11 @@ class DevicePoolExecutor:
f.result()
logger.info("job %s finished successfully", key)
except Exception as err:
logger.warn("job %s failed with an error: %s", key, format_exception(type(err), err, err.__traceback__))
logger.warn(
"job %s failed with an error: %s",
key,
format_exception(type(err), err, err.__traceback__),
)
future.add_done_callback(job_done)

View File

@ -96,17 +96,17 @@ class ImageParams:
self.prompt = prompt
self.negative_prompt = negative_prompt
self.cfg = cfg
self.steps = steps
self.seed = seed
self.steps = steps
def tojson(self) -> Dict[str, Optional[Param]]:
return {
"model": self.model,
"scheduler": self.scheduler.__name__,
"seed": self.seed,
"prompt": self.prompt,
"cfg": self.cfg,
"negativePrompt": self.negative_prompt,
"cfg": self.cfg,
"seed": self.seed,
"steps": self.steps,
}
@ -175,8 +175,15 @@ class UpscaleParams:
def tojson(self):
return {
"model": self.upscale_model,
"scale": self.scale,
"upscale_model": self.upscale_model,
"correction_model": self.correction_model,
"denoise": self.denoise,
"faces": self.faces,
"face_strength": self.face_strength,
"format": self.format,
"half": self.half,
"outscale": self.outscale,
# TODO: add more
"pre_pad": self.pre_pad,
"scale": self.scale,
"tile_pad": self.tile_pad,
}