[fix] Add lstrip to pipeline model output

UKPLab · Jul 18, 2024 · 99a6cfc · 99a6cfc
1 parent 573e264
commit 99a6cfc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/llm_roleplay/VERSION b/llm_roleplay/VERSION
@@ -1 +1 @@
-2.0.6
+2.0.7
diff --git a/llm_roleplay/models/model_pipeline.py b/llm_roleplay/models/model_pipeline.py
@@ -127,12 +127,11 @@ def generate(self, prompt: str, generate_cfg):
                 turn_response = turn_response.split(self_reply_token)[0]
                 self.aim_run["num_self_replies"] += 1
 
+        turn_response = turn_response.lstrip()
         model_output_template = self.conv_template.model_output.replace(
             self.spec_tokens.model_answer, turn_response
         )
 
-        # del output_tokenized
-
         return turn_response, model_output_template
 
     def update_history(self, prompt, output_extract):