SakanaAI
/

EvoLLM-JP-v1-7B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

mkshing commited on Mar 13, 2024

Commit

2b5e874

·

verified ·

1 Parent(s): 5bdbc1e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 # 1. load model
 device = "cuda" if torch.cuda.is_available() else "CPU"
-repo_id = "SakanaAI/EvoLLM-JP"
 model = AutoModelForCausalLM.from_pretrained(repo_id, torch_dtype="auto")
 tokenizer = AutoTokenizer.from_pretrained(repo_id)
 model.to(device)

 # 1. load model
 device = "cuda" if torch.cuda.is_available() else "CPU"
+repo_id = "SakanaAI/EvoLLM-v1-JP-7B"
 model = AutoModelForCausalLM.from_pretrained(repo_id, torch_dtype="auto")
 tokenizer = AutoTokenizer.from_pretrained(repo_id)
 model.to(device)