updated the max token

app.py

@@ -173,7 +173,7 @@ def _generate_batch(prompts: list[str]) -> list[str]:
     with torch.no_grad():
         out = _model.generate(
             **enc,
-            max_new_tokens=512,   # ≈ 200 words — enough for our tight prompt
+            max_new_tokens=2048,   # ≈ 200 words — enough for our tight prompt
             use_cache=True,
             do_sample=False,
             pad_token_id=_tokenizer.pad_token_id,
@@ -198,7 +198,7 @@ def _generate_sequential(prompts: list[str]) -> list[str]:
         with torch.no_grad():
             out = _model.generate(
                 **enc,
-                max_new_tokens=256,
+                max_new_tokens=1024,
                 use_cache=True,
                 do_sample=False,
                 pad_token_id=_tokenizer.pad_token_id,

commitlens.py

@@ -307,22 +307,22 @@ def build_prompts(ctx: CommitContext) -> list[str]:
             sections.append("=== BEFORE CODE ===\n(file did not exist)\n")
 
         # After content
-        if fc.after_content is not None:
-            sections.append(
-                "=== AFTER CODE ===\n"
-                f"{fc.after_content}\n"
-            )
-        else:
-            sections.append("=== AFTER CODE ===\n(file was deleted)\n")
-
-        # Diff patch
-        # if fc.patch:
+        # if fc.after_content is not None:
         #     sections.append(
-        #         "=== DIFF ===\n"
-        #         f"{fc.patch}\n"
+        #         "=== AFTER CODE ===\n"
+        #         f"{fc.after_content}\n"
         #     )
         # else:
-        #     sections.append("=== DIFF ===\n(no patch available)\n")
+        #     sections.append("=== AFTER CODE ===\n(file was deleted)\n")
+
+        # Diff patch
+        if fc.patch:
+            sections.append(
+                "=== DIFF ===\n"
+                f"{fc.patch}\n"
+            )
+        else:
+            sections.append("=== DIFF ===\n(no patch available)\n")
 
         prompts.append("\n".join(sections))