fixed formatting

nod-ai · Feb 12, 2025 · c9cbdfd · c9cbdfd
1 parent 927cfe3
commit c9cbdfd
Showing 1 changed file with 12 additions and 12 deletions.
diff --git 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):