VertexAICustomTrainingJob does not run with accelerator attached

I have a created a PR in the prefect_gcp repository (Add accelerator_count for VertexAICustomTrainingJob by jeremy-thomas-roc · Pull Request #174 · PrefectHQ/prefect-gcp · GitHub) to remedy this, as it is fairly straightforward, but this is a blocking issue for using this infrastructure block. If anyone has a quick workaround until this can be tested and merged, I would appreciate it.