Update debug_ee.py

debug_ee.py

@@ -105,13 +105,15 @@ def run_check(original_name, ee_name, seed, prompt="Hello, how are you?"):
 
     print(f"\n{'='*60}\n")
     return embed_match and logit_match
-
-
-#if __name__ == "__main__":
-parser = argparse.ArgumentParser()
-parser.add_argument("--original", required=True)
-parser.add_argument("--ee",       required=True)
-parser.add_argument("--seed",     type=int, required=True)
-parser.add_argument("--prompt",   default="Hello, how are you?")
-args = parser.parse_args()
-run_check(args.original, args.ee, args.seed, args.prompt)
+original_name='Qwen/Qwen3-0.6B'
+ee_name = 'broadfield-dev/Qwen3-0.6B-dp-ee'
+seed = '424242'
+run_check(original_name, ee_name, seed, prompt="Hello, how are you?")
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--original", required=True)
+    parser.add_argument("--ee",       required=True)
+    parser.add_argument("--seed",     type=int, required=True)
+    parser.add_argument("--prompt",   default="Hello, how are you?")
+    args = parser.parse_args()
+    run_check(args.original, args.ee, args.seed, args.prompt)