import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("llm-jp/llm-jp-3-1.8b")
model = AutoModelForCausalLM.from_pretrained("llm-jp/llm-jp-3-1.8b", device_map="auto", torch_dtype=torch.bfloat16)
text = "自然言語処理とは何か"
tokenized_input = tokenizer.encode(text, add_special_tokens=False, return_tensors="pt").to(model.device)
with torch.no_grad():
output = model.generate(
tokenized_input,
max_new_tokens=100,
do_sample=True,
top_p=0.95,
temperature=0.7,
repetition_penalty=1.05,
)[0]
print(tokenizer.decode(output))
Model Details
Model type:
Transformer-based Language Model
Total seen tokens:
2.1T
Params
Layers
Hidden size
Heads
Context length
Embedding parameters
Non-embedding parameters
1.8b
24
2048
16
4096
407,896,064
1,459,718,144
3.7b
28
3072
24
4096
611,844,096
3,171,068,928
13b
40
5120
40
4096
1,019,740,160
12,688,184,320
Tokenizer
The tokenizer of this model is based on
huggingface/tokenizers
Unigram byte-fallback model.
The vocabulary entries were converted from
llm-jp-tokenizer v3.0
.
Please refer to
README.md
of
llm-jp-tokenizer
for details on the vocabulary construction procedure (the pure SentencePiece training does not reproduce our vocabulary).
Datasets
Pre-training
The models have been pre-trained using a blend of the following datasets.
The models released here are in the early stages of our research and development and have not been tuned to ensure outputs align with human intent and safety considerations.
llm-jp-3-1.8b huggingface.co is an AI model on huggingface.co that provides llm-jp-3-1.8b's model effect (), which can be used instantly with this llm-jp llm-jp-3-1.8b model. huggingface.co supports a free trial of the llm-jp-3-1.8b model, and also provides paid use of the llm-jp-3-1.8b. Support call llm-jp-3-1.8b model through api, including Node.js, Python, http.
llm-jp-3-1.8b huggingface.co is an online trial and call api platform, which integrates llm-jp-3-1.8b's modeling effects, including api services, and provides a free online trial of llm-jp-3-1.8b, you can try llm-jp-3-1.8b online for free by clicking the link below.
llm-jp llm-jp-3-1.8b online free url in huggingface.co:
llm-jp-3-1.8b is an open source model from GitHub that offers a free installation service, and any user can find llm-jp-3-1.8b on GitHub to install. At the same time, huggingface.co provides the effect of llm-jp-3-1.8b install, users can directly use llm-jp-3-1.8b installed effect in huggingface.co for debugging and trial. It also supports api for free installation.