Update debug_ee.py
Browse files- 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 |
-
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)'''
|