Adapted the rest of the scripts to the PEP8 standard

Hukasx0 · Nov 20, 2023 · be7421a · be7421a
1 parent 526e06a
commit be7421a
Show file tree

Hide file tree

Showing 4 changed files with 730 additions and 244 deletions.
diff --git a/app/main-mistral-webui.py b/app/main-mistral-webui.py
@@ -73,7 +73,14 @@ def load_models():
             "temperature": 0.8,
             "context_length": 8192,
             "gpu_layers": gpu_layers,
-            "stop": ["/s", "</s>", "<s>", "[INST]", "[/INST]", "<|im_end|>"],
+            "stop": [
+                "/s",
+                "</s>",
+                "<s>",
+                "[INST]",
+                "[/INST]",
+                "<|im_end|>"
+            ],
         },
     )
 

diff --git a/app/main-mistral.py b/app/main-mistral.py
@@ -84,7 +84,14 @@ def prepare_llm():
             "temperature": 0.8,
             "context_length": 8192,
             "gpu_layers": gpu_layers,
-            "stop": ["/s", "</s>", "<s>", "[INST]", "[/INST]", "<|im_end|>"],
+            "stop": [
+                "/s",
+                "</s>",
+                "<s>",
+                "[INST]",
+                "[/INST]",
+                "<|im_end|>"
+            ],
         },
     )