llama2.mojo:519:15: error: cannot pass 'fn[Int](j = Int) capturing -> None' value, parameter expected 'Int' vectorize[_acc, nelts](size)

[agbarbosa]

Hello team,

I am trying to run the example mojo command line:

mojo llama2.mojo stories15M.bin -s 100 -n 256 -t 0.5 -i "Mojo is a language"

and it is returning the error:

llama2.mojo:519:15: error: cannot pass 'fn[Int](j = Int) capturing -> None' value, parameter expected 'Int' vectorize[_acc, nelts](size)

$ mojo --version
mojo 0.7.0 (af002202)

Could someone please help and check if there's something wrong in the source code? I am new to the language and starting to learn it.

[junweizeng]

Hi @agbarbosa, maybe you need to update the mojo to the newest. I compiled the newest llama2.mojo under the latest mojo and there were no errors.

$ mojo --version
mojo 24.1.0 (55ec12d6)
$ mojo llama2.mojo stories15M.bin -s 100 -n 256 -t 0.5 -i "Mojo is a language"
num parallel workers: 16  SIMD width: 32
Total bytes read: 60816028 Estimated checkpoint size:  57 MB
n layers: 6 | vocab size: 32000
Mojo is a language that people like to talk. Heps when he is happy, when he is sad, when he is bored. He loves to play and sing, but he also likes to be with his friends.
One day, Mojo is playing with his friends in the park. He sees a big, brown dog running and barking. He thinks the dog is friendly, but he is also scared of him. He runs to his friends and tells them about the dog.
His friends are not scared of the dog. They think the dog is big and strong. They say it is a good dog, but Mojo is afraid. He thinks the dog is a bad dog, like him. He hides behind a tree and waits for his friends to go away.
His friends see that Mojo is not scared of the dog. They think he is a good dog, not a bad dog. They want to play with him, but Mojo does not want to play with them. He thinks maybe the dog is not a bad dog, just a big, brown dog. He thinks maybe he is a good dog, but not a bad dog. He is a good dog, not a bad
achieved tok/s:  338.64541832669323

[junweizeng]

Btw, the llama2.mojo I used is as follows:

$ pwd
/path/to/llama2.mojo
$ git show
commit 3154ab59ce98f70e2e9ce8f162410ecc97a8b066 (HEAD -> master, origin/master, origin/HEAD)