Current section

Files

Jump to
viva_tensor include viva_tensor@models@llama_LlamaModel.hrl
Raw

include/viva_tensor@models@llama_LlamaModel.hrl

-record(llama_model, {
embedding :: viva_tensor@nn@embedding:embedding(),
blocks :: list(viva_tensor@models@llama:llama_block()),
final_norm :: viva_tensor@nn@norm:rms_norm(),
output_proj :: viva_tensor@tensor:tensor()
}).