Predicting Distance matrix with large language models