Commit
•
c38f315
1
Parent(s):
b997ba9
Update README.md
Browse files
README.md
CHANGED
@@ -107,4 +107,17 @@ tokenizer.decode(outputs[0][input_ids.shape[-1]:])
|
|
107 |
### Training Infrastructure
|
108 |
|
109 |
- **Hardware**: The model was trained on single node of an Amazon EC2 trn1.32xlarge instance.
|
110 |
-
- **Software**: We use code based on [neuronx-nemo-megatron](https://github.com/aws-neuron/neuronx-nemo-megatron).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
107 |
### Training Infrastructure
|
108 |
|
109 |
- **Hardware**: The model was trained on single node of an Amazon EC2 trn1.32xlarge instance.
|
110 |
+
- **Software**: We use code based on [neuronx-nemo-megatron](https://github.com/aws-neuron/neuronx-nemo-megatron).
|
111 |
+
|
112 |
+
## Citation
|
113 |
+
|
114 |
+
```
|
115 |
+
@misc {karakuri_lm_8x7b_chat_v01,
|
116 |
+
author = { {KARAKURI} {I}nc. },
|
117 |
+
title = { {KARAKURI} {LM} 7{B} {APM} v0.1 },
|
118 |
+
year = { 2024 },
|
119 |
+
url = { https://huggingface.co/karakuri-ai/karakuri-lm-7b-apm-v0.1 },
|
120 |
+
publisher = { Hugging Face },
|
121 |
+
journal = { Hugging Face repository },
|
122 |
+
}
|
123 |
+
```
|