Fixed load_model in test.py

Gadersd / llama2-burn

Llama2 LLM ported to Rust burn

MIT License

272 stars 17 forks source link

Closed filopedraz closed 1 year ago

filopedraz commented 1 year ago

Closes #5

Llama team consolidated the weights in a single file for what concerns llama-2-7b-chat model. Current test.py script is not working.

Gadersd commented 1 year ago

The issue was that I accidentally modified test.py to use safetensors instead of .pth files.