gemma_2b

torchtune.models.gemma.gemma_2b() → TransformerDecoder

用于创建 Gemma 2B 模型的构建器，该模型使用来自以下位置的默认 2B 参数值初始化： https://blog.google/technology/developers/gemma-open-models/

返回值:

Gemma 2B 模型的实例化

返回类型:

TransformerDecoder