Skywork
/

Skywork-Reward-Llama-3.1-8B

Text Classification

text-generation-inference

Model card Files Files and versions Community

chrisliu298 commited on Sep 7, 2024

Commit

7b27ec2

·

verified ·

1 Parent(s): 0f83530

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,7 +70,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 # Load model and tokenizer
 device = "cuda:0"
-model_name = "Skywork/Skywork-Reward-Gemma-2-27B"
 rm = AutoModelForSequenceClassification.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,

 # Load model and tokenizer
 device = "cuda:0"
+model_name = "Skywork/Skywork-Reward-Llama-3.1-8B"
 rm = AutoModelForSequenceClassification.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,