fix formatting.

nod-ai · Dec 18, 2023 · 59153b2 · 59153b2
1 parent 5813043
commit 59153b2
Show file tree

Hide file tree

Showing 7 changed files with 134 additions and 99 deletions.
diff --git a/apps/shark_studio/api/initializers.py b/apps/shark_studio/api/initializers.py
@@ -8,10 +8,10 @@
 
 from apps.shark_studio.modules.timer import startup_timer
 from apps.shark_studio.web.utils.tmp_configs import (
-        config_tmp,
-        clear_tmp_mlir,
-        clear_tmp_imgs,
-    )
+    config_tmp,
+    clear_tmp_mlir,
+    clear_tmp_imgs,
+)
 
 
 def imports():
@@ -57,6 +57,7 @@ def initialize():
     from apps.shark_studio.web.utils.file_utils import (
         create_checkpoint_folders,
     )
+
     # Create custom models folders if they don't exist
     create_checkpoint_folders()
 

diff --git a/apps/shark_studio/api/llm.py b/apps/shark_studio/api/llm.py
@@ -57,7 +57,8 @@ def __init__(
         self.use_system_prompt = use_system_prompt
         self.global_iter = 0
         if os.path.exists(self.vmfb_name) and (
-            external_weights is None or os.path.exists(str(self.external_weight_file))
+            external_weights is None
+            or os.path.exists(str(self.external_weight_file))
         ):
             self.iree_module_dict = dict()
             (