Security vulnerability

[sarthakpati]

According to dependabot:

Link to patch: https://github.com/pytorchlightning/pytorch-lightning/commit/8b7a12c52e52a06408e9231647839ddb4665e8ae

PyTorch lightning 1.6.1 is already out [ref] which should contain this fix, but making any change in dependencies would require testing.

[Geeks-Sid]

yeah, and i have seen 1.6.1 break my code, so i might need to rewrite multiple things. Keeping it open for now, will come back later.

[sarthakpati]

Getting addressed in #19.