From c9cbdfd0fdc9a721358529430ac63be578a11db5 Mon Sep 17 00:00:00 2001 From: daveliddell Date: Wed, 12 Feb 2025 15:39:08 -0800 Subject: [PATCH] fixed formatting --- .../shortfin_apps/sd/components/service.py | 24 +++++++++---------- 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/shortfin/python/shortfin_apps/sd/components/service.py b/shortfin/python/shortfin_apps/sd/components/service.py index 33eb71430..0c8c2d043 100644 --- a/shortfin/python/shortfin_apps/sd/components/service.py +++ b/shortfin/python/shortfin_apps/sd/components/service.py @@ -155,11 +155,11 @@ def start(self): for worker_idx, worker in enumerate(self.workers): self.inference_functions[worker_idx]["encode"] = {} for bs in self.model_params.clip_batch_sizes: - self.inference_functions[worker_idx]["encode"][ - bs - ] = self.inference_programs[worker_idx]["clip"][ - f"{self.model_params.clip_module_name}.encode_prompts" - ] + self.inference_functions[worker_idx]["encode"][bs] = ( + self.inference_programs[worker_idx]["clip"][ + f"{self.model_params.clip_module_name}.encode_prompts" + ] + ) self.inference_functions[worker_idx]["denoise"] = {} for bs in self.model_params.unet_batch_sizes: self.inference_functions[worker_idx]["denoise"][bs] = { @@ -178,15 +178,15 @@ def start(self): } self.inference_functions[worker_idx]["decode"] = {} for bs in self.model_params.vae_batch_sizes: - self.inference_functions[worker_idx]["decode"][ - bs - ] = self.inference_programs[worker_idx]["vae"][ - f"{self.model_params.vae_module_name}.decode" - ] - #self.batcher.launch() + self.inference_functions[worker_idx]["decode"][bs] = ( + self.inference_programs[worker_idx]["vae"][ + f"{self.model_params.vae_module_name}.decode" + ] + ) + # self.batcher.launch() def shutdown(self): - #self.batcher.shutdown() + # self.batcher.shutdown() pass def __repr__(self):