What is the most suitable PyTorch version for running this model? Also, can FLA3 in the model be replaced with FL2?