reaperdoesntknow commited on
Commit
735ce1c
·
verified ·
1 Parent(s): 42bd6ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -1
README.md CHANGED
@@ -37,7 +37,13 @@ model_id = "reaperdoesntknow/Qemma-GEI"
37
  tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=True)
38
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16).eval()
39
 
40
- text = "I notice that the sum involves the absolute values of three linear expressions of x."
 
 
 
 
 
 
41
  inputs = tokenizer(text, return_tensors="pt", max_length=64, padding='max_length', truncation=True)
42
  inputs = {k: v.to(model.device) for k, v in inputs.items()}
43
 
 
37
  tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=True)
38
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16).eval()
39
 
40
+ text = (
41
+ "<|user|>"
42
+ "What makes the sky blue?."
43
+ "<|assistant|>"
44
+ "<think><reasoning_step>"
45
+ )
46
+
47
  inputs = tokenizer(text, return_tensors="pt", max_length=64, padding='max_length', truncation=True)
48
  inputs = {k: v.to(model.device) for k, v in inputs.items()}
49