Existing Llama Guard recipes uses "O" rather than "S" as category code prefix.

[jingzhaoou]

System Info

Nvidia driver version: 550.90.12
cuDNN version: Probably one of the following:
/usr/lib/x86_64-linux-gnu/libcudnn.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_adv.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_cnn.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_engines_precompiled.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_engines_runtime_compiled.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_graph.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_heuristic.so.9.5.1
/usr/lib/x86_64-linux-gnu/libcudnn_ops.so.9.5.1
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

Information

• The official example scripts
• My own modified scripts

🐛 Describe the bug

According to the Llama Guard model card on Hugging Face at here, the category code prefix should be "S". However, when I checked the existing recipe, it uses "O". Wonder if this is a bug. Thanks.

llama-cookbook/src/llama_recipes/datasets/toxicchat_dataset.py

Lines 62 to 63 in ef4a42a

	categories=LLAMA_GUARD_3_CATEGORY,
	category_code_prefix="O")

Error logs

Not Available

Expected behavior

Not Available