Fix CI and install docs (#3821)

2025-02-24 16:17:38 -08:00
parent 62bbd34393
commit d7934cde45
10 changed files with 36 additions and 42 deletions
--- a/test/srt/test_hidden_states.py
+++ b/test/srt/test_hidden_states.py
@@ -62,7 +62,7 @@ class TestHiddenState(unittest.TestCase):
                f"Max diff: {torch.max(torch.abs(hf_out['hidden_states'][-1][0] - sg_hidden_states))}"
            )

-            atol = 0.8 if is_in_ci() else 0.4
+            atol = 0.8
            self.assertTrue(
                torch.allclose(
                    hf_out["hidden_states"][-1][0],