Update README.md
Browse files
README.md
CHANGED
|
@@ -72,7 +72,7 @@ Below is sample code to use the model. Note that the model requires a custom mod
|
|
| 72 |
import torch
|
| 73 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 74 |
|
| 75 |
-
model_path = "inceptionai/
|
| 76 |
|
| 77 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 78 |
|
|
|
|
| 72 |
import torch
|
| 73 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 74 |
|
| 75 |
+
model_path = "inceptionai/jais-family-256m"
|
| 76 |
|
| 77 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 78 |
|