gemma2_9b¶
- torchtune.models.gemma2.gemma2_9b() TransformerDecoder[source]¶
Builder for creating a Gemma2 9B model initialized w/ the default 9b parameter values from: https://github.com/google/gemma_pytorch/blob/main/gemma/config.py
- Returns:
Instantiation of Gemma 9B model
- Return type:
