Fix flake8 issues

Fix minor issues in finetune script

Fix python 3 download script and left-over cuda in attention layer

Initial commit