From d8bb4db800ef1de584b729ced5fbcbf9589bcca9 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Thu, 30 May 2024 03:46:47 -0500 Subject: [PATCH] formatting --- apps/shark_studio/api/sd.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py index e9ef69a9..a0905587 100644 --- a/apps/shark_studio/api/sd.py +++ b/apps/shark_studio/api/sd.py @@ -58,7 +58,6 @@ EMPTY_FLAGS = { } - def load_script(source, module_name): """ reads file source and loads it as a module @@ -108,11 +107,8 @@ class StableDiffusion: self.is_custom = "custom" in self.base_model_id.lower() if self.is_custom: custom_module = load_script( - os.path.join( - get_checkpoints_path("scripts"), - self.base_model_id - ), - "custom_pipeline" + os.path.join(get_checkpoints_path("scripts"), self.base_model_id), + "custom_pipeline", ) self.turbine_pipe = custom_module.StudioPipeline self.model_map = custom_module.MODEL_MAP