nvidia
/

Llama-3.1-Nemotron-70B-Reward-HF

Model card Files Files and versions Community

zhilinw commited on Sep 28

Commit

ea34450

•

1 Parent(s): e1ef9e9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -75,7 +75,7 @@ This code has been tested on Transformers v4.44.0, torch v2.4.0 and 2 A100 80GB
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "Llama-3.1-Nemotron-70B-Reward-HF"
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_name)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "nvidia/Llama-3.1-Nemotron-70B-Reward-HF"
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_name)