upgrade lightning dependency

[ankitade]

• Upgrade lightning to 1.6 to fix installation errors
• Fix some tests since lightning logic to track global step has changed

[facebook-github-bot]

@ankitade has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[facebook-github-bot]

@ankitade merged this pull request in facebookresearch/mmf@c82aace54b5ded70438c4af74450ca0472152c1e.