broadfield-dev commited on
Commit
65fc5c6
·
verified ·
1 Parent(s): 319b3f9

Update debug_ee.py

Browse files
Files changed (1) hide show
  1. debug_ee.py +8 -6
debug_ee.py CHANGED
@@ -105,15 +105,17 @@ def run_check(original_name, ee_name, seed, prompt="Hello, how are you?"):
105
 
106
  print(f"\n{'='*60}\n")
107
  return embed_match and logit_match
108
- original_name='Qwen/Qwen3-0.6B'
109
- ee_name = 'broadfield-dev/Qwen3-0.6B-dp-ee'
110
- seed = '424242'
111
- run_check(original_name, ee_name, seed, prompt="Hello, how are you?")
112
  if __name__ == "__main__":
113
- parser = argparse.ArgumentParser()
 
 
 
 
 
114
  parser.add_argument("--original", required=True)
115
  parser.add_argument("--ee", required=True)
116
  parser.add_argument("--seed", type=int, required=True)
117
  parser.add_argument("--prompt", default="Hello, how are you?")
118
  args = parser.parse_args()
119
- run_check(args.original, args.ee, args.seed, args.prompt)
 
105
 
106
  print(f"\n{'='*60}\n")
107
  return embed_match and logit_match
108
+
 
 
 
109
  if __name__ == "__main__":
110
+
111
+ original_name='Qwen/Qwen3-0.6B'
112
+ ee_name = 'broadfield-dev/Qwen3-0.6B-dp-ee'
113
+ seed = '424242'
114
+ run_check(original_name, ee_name, seed, prompt="Hello, how are you?")
115
+ '''parser = argparse.ArgumentParser()
116
  parser.add_argument("--original", required=True)
117
  parser.add_argument("--ee", required=True)
118
  parser.add_argument("--seed", type=int, required=True)
119
  parser.add_argument("--prompt", default="Hello, how are you?")
120
  args = parser.parse_args()
121
+ run_check(args.original, args.ee, args.seed, args.prompt)'''