PyTorch doesn’t guarantee forward compatibility, so while your “hack” might work, you might encounter unwanted side effects and issues.
If possible, try to save and load the model in 1.6 or update to 1.7.1 (which would be the better approach).