KeisukeMiyamoto commited on
Commit
832cdb9
·
verified ·
1 Parent(s): 82bea6f

Add model card

Browse files
Files changed (1) hide show
  1. README.md +73 -0
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - ja
4
+ library_name: pytorch
5
+ pipeline_tag: text-generation
6
+ datasets:
7
+ - KeisukeMiyamoto/lambda-corpus
8
+ - KeisukeMiyamoto/SyntheticTextbook-jp
9
+ - KeisukeMiyamoto/SyntheticTalk-jp
10
+ base_model: KeisukeMiyamoto/lambda-1-360m-mid
11
+ base_model_relation: finetune
12
+ tags:
13
+ - lambda
14
+ - pytorch
15
+ - causal-lm
16
+ - text-generation
17
+ - conversational
18
+ - instruction-tuned
19
+ - supervised-fine-tuning
20
+ - decoder-only
21
+ - custom-code
22
+ - grouped-query-attention
23
+ - rotary-position-embedding
24
+ - byte-level-bpe
25
+ ---
26
+
27
+ # lambda-1-360m-it
28
+
29
+ lambda-1-360m-it is an experimental Japanese instruction-tuned language model based on [`lambda-1-360m-mid`](https://huggingface.co/KeisukeMiyamoto/lambda-1-360m-mid) and supervised fine-tuned on a synthetic Japanese conversation corpus.
30
+
31
+ All training code is publicly available at [KeisukeMiyamoto1324/lambda](https://github.com/KeisukeMiyamoto1324/lambda).
32
+
33
+ ## Model Details
34
+
35
+ | Item | Value |
36
+ |---|---:|
37
+ | Parameters | 359.9M |
38
+ | Architecture | Decoder-only Transformer |
39
+ | Context length | 1,024 tokens |
40
+ | Tokenizer | Byte-level BPE |
41
+ | Vocabulary size | 65,536 |
42
+ | Layers | 32 |
43
+ | Hidden size | 960 |
44
+ | Attention heads | 15 |
45
+ | Key-value heads | 5 |
46
+ | FFN size | 4,096 |
47
+
48
+ ## Training Data
49
+
50
+ The base model was pretrained on [`KeisukeMiyamoto/lambda-corpus`](https://huggingface.co/datasets/KeisukeMiyamoto/lambda-corpus) and further pretrained on [`KeisukeMiyamoto/SyntheticTextbook-jp`](https://huggingface.co/datasets/KeisukeMiyamoto/SyntheticTextbook-jp).
51
+
52
+ This model was then supervised fine-tuned on [`KeisukeMiyamoto/SyntheticTalk-jp`](https://huggingface.co/datasets/KeisukeMiyamoto/SyntheticTalk-jp) for 5,000 steps.
53
+
54
+ ## Usage
55
+
56
+ ```bash
57
+ git clone https://github.com/KeisukeMiyamoto1324/lambda.git
58
+ cd lambda
59
+ python3 -m venv venv
60
+ source venv/bin/activate
61
+ pip3 install -r requirements.txt
62
+
63
+ python3 src/inference_it/inference.py \
64
+ --model-dir "KeisukeMiyamoto/lambda-1-360m-it" \
65
+ --prompt "人工知能とは何ですか" \
66
+ --max-new-tokens 128
67
+ ```
68
+
69
+ ## Limitations
70
+
71
+ This model is not safety-aligned. It may generate incorrect, biased, unsafe, or low-quality text.
72
+
73
+ The model was trained primarily on synthetic Japanese text and has not been evaluated on standard benchmarks.