made lp tolorance in fuzzing slightly looser

lens_migration.py

@@ -378,7 +378,7 @@ if __name__ == "__main__":
                 logits_old = tuned_lens_old(a, i)
                 log_ps_new = logits_new.log_softmax(-1)
                 log_ps_old = logits_old.log_softmax(-1)
-                assert (th.allclose(log_ps_new, log_ps_old))
+                assert (th.allclose(log_ps_new, log_ps_old, atol=1e-7))
                 print("js div", js_divergence(log_ps_new, log_ps_old))
     print("Saving new lens to", args.output_dir)
     tuned_lens.to(th.device("cpu")).save(args.output_dir)