Fix Llama3 HF checkpoint converter (#202)

Author: yanboliang

scripts/convert_hf_checkpoint.py

@@ -115,7 +115,7 @@ def permute(w, n_head):
             del final_result[key.replace("wq", "wv")]
     print(f"Saving checkpoint to {checkpoint_dir / 'model.pth'}")
     torch.save(final_result, checkpoint_dir / "model.pth")
-    if 'llama-3' in model_name.lower():
+    if 'llama-3-' in model_name.lower() or 'llama-3.1-' in model_name.lower():
         if 'llama-3.1-405b' in model_name.lower():
             original_dir = checkpoint_dir / "original" / "mp16"
         else: