GenerTeam commited on
Commit
3a5d177
·
verified ·
1 Parent(s): bfd89cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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