Update README.md
Browse files
README.md
CHANGED
|
@@ -33,9 +33,14 @@ from transformers import AutoTokenizer
|
|
| 33 |
import transformers
|
| 34 |
import torch
|
| 35 |
|
| 36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
-
tokenizer = AutoTokenizer.from_pretrained(model, trust_remote_code=True)
|
| 39 |
messages = [
|
| 40 |
{
|
| 41 |
"role": "user",
|
|
@@ -114,9 +119,14 @@ from transformers import AutoTokenizer
|
|
| 114 |
import transformers
|
| 115 |
import torch
|
| 116 |
|
| 117 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
|
| 119 |
-
tokenizer = AutoTokenizer.from_pretrained(model, trust_remote_code=True)
|
| 120 |
messages = [
|
| 121 |
{
|
| 122 |
"role": "user",
|
|
|
|
| 33 |
import transformers
|
| 34 |
import torch
|
| 35 |
|
| 36 |
+
model_name = "opencsg/OpenCSG-R1-Qwen2.5-Math-3B-V1"
|
| 37 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 38 |
+
model_name,
|
| 39 |
+
torch_dtype="auto",
|
| 40 |
+
device_map="auto"
|
| 41 |
+
)
|
| 42 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
| 43 |
|
|
|
|
| 44 |
messages = [
|
| 45 |
{
|
| 46 |
"role": "user",
|
|
|
|
| 119 |
import transformers
|
| 120 |
import torch
|
| 121 |
|
| 122 |
+
model_name = "opencsg/OpenCSG-R1-Qwen2.5-Math-3B-V1"
|
| 123 |
+
model = AutoModelForCausalLM.from_pretrained(
|
| 124 |
+
model_name,
|
| 125 |
+
torch_dtype="auto",
|
| 126 |
+
device_map="auto"
|
| 127 |
+
)
|
| 128 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
| 129 |
|
|
|
|
| 130 |
messages = [
|
| 131 |
{
|
| 132 |
"role": "user",
|