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