Hugging Face's logo

nicolinho
/
QRM-Llama3.1-8B-v2

Safetensors
llama
custom_code
Model card Files Community
1
nicolinho commited on
Commit
1b1d3a1
·
verified ·
1 Parent(s): e2647a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ Apart from this, it has been trained exactly as described in the tech report.
27
  import torch
28
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
29
  device = "cuda"
30
- path = "/home/dorka/projects/24/qrm_hf/QRM-Llama3.1-8B-v2"
31
  model = AutoModelForSequenceClassification.from_pretrained(path, torch_dtype=torch.bfloat16, device_map=device, trust_remote_code=True)
32
  tokenizer = AutoTokenizer.from_pretrained(path, use_fast=True)
33
  # We load a random sample from the validation set of the HelpSteer dataset
 
27
  import torch
28
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
29
  device = "cuda"
30
+ path = "nicolinho/QRM-Llama3.1-8B-v2"
31
  model = AutoModelForSequenceClassification.from_pretrained(path, torch_dtype=torch.bfloat16, device_map=device, trust_remote_code=True)
32
  tokenizer = AutoTokenizer.from_pretrained(path, use_fast=True)
33
  # We load a random sample from the validation set of the HelpSteer dataset