This commit is contained in:
Nicolas Patry 2024-02-13 21:52:02 +00:00
parent b9ac720d1e
commit 1ffc3a03c8

View File

@ -63,4 +63,4 @@ async def test_mamba_load(fused_kernel_mamba, generate_load, generous_response_s
assert all([r.generated_text == responses[0].generated_text for r in responses]) assert all([r.generated_text == responses[0].generated_text for r in responses])
assert responses[0].generated_text == "\n\nDeep learning is a new type of machine" assert responses[0].generated_text == "\n\nDeep learning is a new type of machine"
assert responses == response_snapshot assert responses == generous_response_snapshot