gemma_2b¶
- torchtune.models.gemma.gemma_2b() TransformerDecoder [source]¶
Builder for creating a Gemma 2B model initialized w/ the default 2b parameter values from: https://blog.google/technology/developers/gemma-open-models/
- Returns:
Instantiation of Gemma 2B model
- Return type: