Upload 137 files

hugging/td_fuse/validate.py

@@ -217,10 +217,8 @@ def test_thinking_mode(
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
-            max_new_tokens=300,
-            do_sample=True,
-            temperature=0.7,
-            top_p=0.9,
+            max_new_tokens=800,
+            do_sample=False,
         )
 
     # Decode only the NEW tokens (skip the prompt)

hugging/td_lang/compiler.py

@@ -1011,6 +1011,10 @@ DO NOT EDIT - regenerate from the .td file instead.
         self._emit('           "Provide a formal argument for: {problem}"],')
         self._emit('"factual": ["Answer with citations: {problem}",')
         self._emit('            "List 3 verified facts about: {problem}"],')
+        self._emit('"creativity": ["Think of an original approach to: {problem}",')
+        self._emit('               "Brainstorm 5 creative solutions for: {problem}"],')
+        self._emit('"instruction": ["Follow these instructions precisely: {problem}",')
+        self._emit('                "Complete this task exactly as described: {problem}"],')
         self._indent -= 1
         self._emit("}")
         self._emit("")
@@ -1053,6 +1057,22 @@ DO NOT EDIT - regenerate from the .td file instead.
         self._emit("'What causes tides on Earth?',")
         self._indent -= 1
         self._emit("],")
+        self._emit("'creativity': [")
+        self._indent += 1
+        self._emit("'Design a new board game that teaches fractions to kids',")
+        self._emit("'Invent a product that solves a common kitchen problem',")
+        self._emit("'Write a short story where time flows backwards',")
+        self._emit("'Propose 3 unconventional uses for a paperclip',")
+        self._indent -= 1
+        self._emit("],")
+        self._emit("'instruction': [")
+        self._indent += 1
+        self._emit("'Write exactly 3 sentences about dogs. Each must start with a different letter.',")
+        self._emit("'List the planets in order from the sun. Format each as: N. Name - one interesting fact.',")
+        self._emit("'Translate this to formal English then to casual English: gonna grab some grub',")
+        self._emit("'Summarize photosynthesis in exactly 25 words.',")
+        self._indent -= 1
+        self._emit("],")
         self._indent -= 1
         self._emit("}")
         self._emit("")
@@ -1097,7 +1117,7 @@ DO NOT EDIT - regenerate from the .td file instead.
         self._indent += 1
         self._emit("domain = random.choice(weak_topics)")
         self._emit("problem = make_problem(domain)")
-        self._emit("template = random.choice(domain_templates[domain])")
+        self._emit("template = random.choice(domain_templates.get(domain, ['Solve this problem step by step: {problem}']))")
         self._emit('prompt = template.format(problem=problem)')
         self._emit('inputs = tok(prompt, return_tensors="pt").to(model.device)')
         self._emit("with torch.no_grad():")