Security vulnerability

[sarthakpati]

According to dependabot:

Link to patch: Lightning-AI/pytorch-lightning@8b7a12c

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.