Update README.md
Browse files
README.md
CHANGED
|
@@ -139,7 +139,7 @@ class aaronSystemOne(nn.Module):
|
|
| 139 |
# Note: Ensure modeling_jev.py is in your working directory
|
| 140 |
from modeling_jev import aaronEngine, Choice, Noul, Score
|
| 141 |
|
| 142 |
-
MODEL_ID = "
|
| 143 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 144 |
|
| 145 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
|
|
|
| 139 |
# Note: Ensure modeling_jev.py is in your working directory
|
| 140 |
from modeling_jev import aaronEngine, Choice, Noul, Score
|
| 141 |
|
| 142 |
+
MODEL_ID = "yothinS/Aaron"
|
| 143 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 144 |
|
| 145 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|