Update README.md
Browse files
README.md
CHANGED
|
@@ -84,7 +84,7 @@ Here is a simple Python script to load the model and generate text interactively
|
|
| 84 |
import torch
|
| 85 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 86 |
|
| 87 |
-
MODEL_ID = "
|
| 88 |
|
| 89 |
def main():
|
| 90 |
print("Loading Pebble 10M...")
|
|
|
|
| 84 |
import torch
|
| 85 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 86 |
|
| 87 |
+
MODEL_ID = "basically-ai/Pebble-10M-Base"
|
| 88 |
|
| 89 |
def main():
|
| 90 |
print("Loading Pebble 10M...")
|