Update README.md
Browse files
README.md
CHANGED
|
@@ -23,8 +23,8 @@ import torch
|
|
| 23 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 24 |
|
| 25 |
# Load the tokenizer and model.
|
| 26 |
-
tokenizer = AutoTokenizer.from_pretrained("GENERator-eukaryote-3b-base", trust_remote_code=True)
|
| 27 |
-
model = AutoModelForCausalLM.from_pretrained("GENERator-eukaryote-3b-base")
|
| 28 |
config = model.config
|
| 29 |
|
| 30 |
max_length = config.max_position_embeddings
|
|
|
|
| 23 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 24 |
|
| 25 |
# Load the tokenizer and model.
|
| 26 |
+
tokenizer = AutoTokenizer.from_pretrained("GenerTeam/GENERator-eukaryote-3b-base", trust_remote_code=True)
|
| 27 |
+
model = AutoModelForCausalLM.from_pretrained("GenerTeam/GENERator-eukaryote-3b-base")
|
| 28 |
config = model.config
|
| 29 |
|
| 30 |
max_length = config.max_position_embeddings
|