Update README.md

README.md

@@ -3,11 +3,11 @@ license: apache-2.0
 ---
 # Dataset
 
-Japanese subset of the mC4 dataset
+Japanese subset of the [mC4](https://huggingface.co/datasets/mc4) dataset
 
 # Training
 
-Trained for 3000 steps on top of the MPT 7b checkpoint mosaicml/mpt-7b
+Trained for 3000 steps on top of the MPT 7b checkpoint [mosaicml/mpt-7b](https://huggingface.co/mosaicml/mpt-7b)
 
 # How to load
 
@@ -17,7 +17,7 @@ Before running this model, please install the following pip package:
 pip install einops
 ```
 
-To run this model, you may need to load it in a lower precision in order for it to fit onto your GPU. We found for a T4 GPU, it requires loading the model in 8-bit precision. To load the model in 8-bit or 4-bit, please install the following pip packages:
+To run this model, you may need to load it in a lower precision in order for it to fit onto your GPU. We found for a T4 GPU, it requires loading the model in 8-bit precision. To load the model in 8-bit, please install the following pip packages:
 
 ```bash
 pip install bitsandbytes accelerate
@@ -61,25 +61,6 @@ model = AutoModelForCausalLM.from_pretrained(
 
 </details>
 
-<details>
-<summary><b>In 4 bit</b></summary>
-
-
-
-```python
-from transformers import AutoModelForCausalLM
-
-model_name = "lightblue/japanese-mpt-7b"
-model = AutoModelForCausalLM.from_pretrained(
-    model_name,
-    torch_dtype='auto',
-    load_in_4bit=True,
-    trust_remote_code=True
-)
-```
-
-</details>
-
 
 # How to use
 ```python