Update README.md

README.md

@@ -42,15 +42,26 @@ It includes options for many training efficiency features such as [FlashAttentio
 
 ```python
 import transformers
-model = transformers.AutoModelForCausalLM.from_pretrained('mosaicml/mpt-7b-storywriter', trust_remote_code=True)
+model = transformers.AutoModelForCausalLM.from_pretrained(
+  'mosaicml/mpt-7b-storywriter',
+  trust_remote_code=True
+)
 ```
 
 To use the optimized [triton implementation](https://github.com/openai/triton) of FlashAttention, you can load the model with `attn_impl='triton'` and move the model to `bfloat16`:
 ```python
-config = transformers.AutoConfig.from_pretrained('mosaicml/mpt-7b-storywriter', trust_remote_code=True)
+config = transformers.AutoConfig.from_pretrained(
+  'mosaicml/mpt-7b-storywriter',
+  trust_remote_code=True
+)
 config.attn_config['attn_impl'] = 'triton'
 
-model = transformers.AutoModelForCausalLM.from_pretrained('mosaicml/mpt-7b-storywriter', config=config, torch_dtype=torch.bfloat16, trust_remote_code=True)
+model = transformers.AutoModelForCausalLM.from_pretrained(
+  'mosaicml/mpt-7b-storywriter',
+  config=config,
+  torch_dtype=torch.bfloat16,
+  trust_remote_code=True
+)
 model.to(device='cuda:0')
 ```
 
@@ -58,9 +69,16 @@ Although the model was trained with a sequence length of 2048 and finetuned with
 ALiBi enables users to increase the maximum sequence length during finetuning and/or inference. For example:
 
 ```python
-config = transformers.AutoConfig.from_pretrained('mosaicml/mpt-7b-storywriter', trust_remote_code=True)
+config = transformers.AutoConfig.from_pretrained(
+  'mosaicml/mpt-7b-storywriter',
+  trust_remote_code=True
+)
 config.update({"max_seq_len": 83968})
-model = transformers.AutoModelForCausalLM.from_pretrained('mosaicml/mpt-7b-storywriter', config=config, trust_remote_code=True)
+model = transformers.AutoModelForCausalLM.from_pretrained(
+  'mosaicml/mpt-7b-storywriter',
+  config=config,
+  trust_remote_code=True
+)
 ```
 
 This model was trained with the [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) tokenizer.