llama-2-jax
llama-2-jax copied to clipboard
13B parameter model
Thanks for the effort to port Llama 2 to jax. After looking further in the code, it seems that the config for the 13B model is missing.
Relevant file:
lib/model/ModelConfig.py