codellama-playground

Running

lvwerra HF Staff commited on Aug 25, 2023

Commit

4206e00

1 Parent(s): f664b2f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ def generate(
         if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):
             if fim_mode:
                 output += suffix
-            print("output", output)
             else:
                 return output
         else:

         if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):
             if fim_mode:
                 output += suffix
+                print("output", output)
             else:
                 return output
         else: