Tree: Format

Signed-off-by: kingbri <bdashore3@proton.me>

common/sampling.py

@@ -125,8 +125,7 @@ class BaseSamplerRequest(BaseModel):
             "rep_pen_range",
         ),
         description=(
-            "Aliases: repetition_range, repetition_penalty_range, "
-            "rep_pen_range"
+            "Aliases: repetition_range, repetition_penalty_range, " "rep_pen_range"
         ),
     )
 

common/transformers_utils.py

@@ -47,9 +47,7 @@ class HuggingFaceConfig(BaseModel):
         """Create an instance from a generation config file."""
 
         hf_config_path = model_directory / "config.json"
-        with open(
-            hf_config_path, "r", encoding="utf8"
-        ) as hf_config_json:
+        with open(hf_config_path, "r", encoding="utf8") as hf_config_json:
             hf_config_dict = json.load(hf_config_json)
             return self.model_validate(hf_config_dict)
 

endpoints/Kobold/types/generation.py

@@ -27,7 +27,7 @@ class GenerateRequest(BaseSamplerRequest):
         if self.use_default_badwordsids:
             bad_words_ids = unwrap(
                 model.container.generation_config.bad_words_ids,
-                model.container.hf_config.get_badwordsids()
+                model.container.hf_config.get_badwordsids(),
             )
 
             if bad_words_ids: