I want to use the Flowformer on a system without GPU but the code uses CUDA. Can someone explain what I have to change to run the train.sh?