Describe the feature

[rank0]: NotImplementedError: Auto policy for Gemma2ForCausalLM (transformers.models.gemma2.modeling_gemma2.Gemma2ForCausalLM) is not implemented

can you please tell me how to support gemma2 for Tensor Parallelism?
or do you have a plan for it?