Update tests/test_model.py

Commit

4 years ago

Update tests/test_model.py Co-authored-by: Niklas Muennighoff <n.muennighoff@gmail.com>

References

Author

thomasw21

Parents

Megatron-DeepSpeed 16c556c0 - Update tests/test_model.py