GenerTeam
/

GENERator-eukaryote-1.2b-base

Text Generation

text-generation-inference

Model card Files Files and versions

U-rara commited on Feb 11, 2025

Commit

552157b

·

1 Parent(s): 6dc7122

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -24,8 +24,8 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the tokenizer and model.
-tokenizer = AutoTokenizer.from_pretrained("GENERator-eukaryote-1.2b-base", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("GENERator-eukaryote-1.2b-base")
 config = model.config
 max_length = config.max_position_embeddings

 from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the tokenizer and model.
+tokenizer = AutoTokenizer.from_pretrained("GenerTeam/GENERator-eukaryote-1.2b-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("GenerTeam/GENERator-eukaryote-1.2b-base")
 config = model.config
 max_length = config.max_position_embeddings