From 310337daa0d05290dd0d9ec7f96b22362e83112b Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 30 Apr 2025 08:52:26 -0700
Subject: [PATCH] Update synthetic.py

---
 unsloth/dataprep/synthetic.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py
index 4a26da87ab..f3c3fc9330 100644
--- a/unsloth/dataprep/synthetic.py
+++ b/unsloth/dataprep/synthetic.py
@@ -232,6 +232,8 @@ class SyntheticDataKit:
             os.makedirs(os.path.join(output_folder, path), exist_ok = True)
         pass
 
+        self.max_generation_tokens = max_generation_tokens
+
         config = synthetic_qa_config\
             .replace("{data_output_location}", str(output_folder))\
             .replace("{model_name}", str(self.model_name))\