fix: linter format (#4832)

GitOrigin-RevId: c5195206f02b13286677421bbe8c27f8df6bebc9
pathwaycom · Oct 20, 2023 · a65e20e · a65e20e
1 parent 760b946
commit a65e20e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/llm_app/model_wrappers/huggingface_wrapper/pipelines.py b/llm_app/model_wrappers/huggingface_wrapper/pipelines.py
@@ -118,7 +118,9 @@ def __call__(self, text, **kwargs):
             },
             {"role": "user", "content": text},
         ]
-        prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+        prompt = self.tokenizer.apply_chat_template(
+            messages, tokenize=False, add_generation_prompt=True
+        )
 
         output = self.pipeline(prompt, max_new_tokens=max_new_tokens, **kwargs)
         return output[0]["generated_text"]