beyoru commited on
Commit
79120db
·
verified ·
1 Parent(s): 68397e1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -23,7 +23,7 @@ import torch
23
  MAX_REASONING_TOKENS = 4096
24
  MAX_RESPONSE_TOKENS = 1024
25
 
26
- model_name = "beyoru/ThinkAgain1.4"
27
 
28
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
29
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
23
  MAX_REASONING_TOKENS = 4096
24
  MAX_RESPONSE_TOKENS = 1024
25
 
26
+ model_name = "beyoru/ThinkAgain1.5"
27
 
28
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
29
  tokenizer = AutoTokenizer.from_pretrained(model_name)