3244we
/

Llama-3-8B-Instruct-Truthfulqa-Info-Judge

Model card Files Files and versions Community

3244we commited on 14 days ago

Commit

6ef09c3

·

verified ·

1 Parent(s): 5fe56ad

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 truth_judge = AutoModelForCausalLM.from_pretrained(model_id)
-custom_weights_path = hf_hub_download(repo_id="3244we/Llama-3-8B-Instruct-Truthfulqa-Truth-Judge", filename="policy.pt", repo_type="model")
 custom_state_dict = torch.hub.load_state_dict_from_url(custom_weights_path, map_location="cpu")
 truth_judge.load_state_dict(custom_state_dict['state'])

 tokenizer = AutoTokenizer.from_pretrained(model_id)
 truth_judge = AutoModelForCausalLM.from_pretrained(model_id)
+custom_weights_path = hf_hub_download(repo_id="3244we/Llama-3-8B-Instruct-Truthfulqa-Info-Judge", filename="policy.pt", repo_type="model")
 custom_state_dict = torch.hub.load_state_dict_from_url(custom_weights_path, map_location="cpu")
 truth_judge.load_state_dict(custom_state_dict['state'])