Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -120,7 +120,7 @@ hard one (risking a bad answer) — the direction you want a router to err in.
|
|
| 120 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 121 |
import torch
|
| 122 |
|
| 123 |
-
repo_id = "appriai/gen-router-
|
| 124 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
| 125 |
model = AutoModelForSequenceClassification.from_pretrained(repo_id)
|
| 126 |
|
|
@@ -138,7 +138,7 @@ import onnxruntime as ort
|
|
| 138 |
from transformers import AutoTokenizer
|
| 139 |
from huggingface_hub import hf_hub_download
|
| 140 |
|
| 141 |
-
repo_id = "appriai/gen-router-
|
| 142 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
| 143 |
onnx_path = hf_hub_download(repo_id, "onnx/model.onnx")
|
| 144 |
session = ort.InferenceSession(onnx_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
|
|
|
| 120 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 121 |
import torch
|
| 122 |
|
| 123 |
+
repo_id = "appriai/gen-router-t1"
|
| 124 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
| 125 |
model = AutoModelForSequenceClassification.from_pretrained(repo_id)
|
| 126 |
|
|
|
|
| 138 |
from transformers import AutoTokenizer
|
| 139 |
from huggingface_hub import hf_hub_download
|
| 140 |
|
| 141 |
+
repo_id = "appriai/gen-router-t1"
|
| 142 |
tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
| 143 |
onnx_path = hf_hub_download(repo_id, "onnx/model.onnx")
|
| 144 |
session = ort.InferenceSession(onnx_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|