Update split string in doctest to reflect #28087 (#28135)

This commit is contained in:
amyeroberts
2023-12-19 13:55:09 +00:00
committed by GitHub
parent 5aec50ecaf
commit bd7a356135

View File

@@ -60,7 +60,7 @@ class TestDocLists(unittest.TestCase):
doctext = f.read() doctext = f.read()
doctext = doctext.split( doctext = doctext.split(
"For now, Transformers supports inference and training through SDPA for the following architectures:" "For now, Transformers supports SDPA inference and training for the following architectures:"
)[1] )[1]
doctext = doctext.split("Note that FlashAttention can only be used for models using the")[0] doctext = doctext.split("Note that FlashAttention can only be used for models using the")[0]