nicolinho
/

QRM-Llama3-8B

Model card Files Files and versions Community

nicolinho commited on Sep 24, 2024

Commit

a7fc4f7

·

verified ·

1 Parent(s): 6c6f0bc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ license: llama3
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 device = "cuda"
-path = "RLHFlow/ArmoRM-Llama3-8B-v0.1"
 model = AutoModelForSequenceClassification.from_pretrained(path, device_map=device,
                                trust_remote_code=True, torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(path, use_fast=True)

 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 device = "cuda"
+path = "nicolinho/QRM-Llama3-8B"
 model = AutoModelForSequenceClassification.from_pretrained(path, device_map=device,
                                trust_remote_code=True, torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(path, use_fast=True)