Update README.md

README.md

@@ -57,8 +57,8 @@ Including `torch_dtype=torch.bfloat16` is generally recommended if this type is
 It is also fine to remove it if there is sufficient memory.
 
 ```python
-import torch
 from transformers import pipeline
+import torch
 
 generate_text = pipeline(model="aisquared/chopt-research-1_3b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
 ```
@@ -74,9 +74,9 @@ Alternatively, if you prefer to not use `trust_remote_code=True` you can downloa
 store it alongside your notebook, and construct the pipeline yourself from the loaded model and tokenizer:
 
 ```python
-import torch
 from instruct_pipeline import InstructionTextGenerationPipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
 tokenizer = AutoTokenizer.from_pretrained("aisquared/chopt-research-1_3b", padding_side="left")
 model = AutoModelForCausalLM.from_pretrained("aisquared/chopt-research-1_3b", device_map="auto", torch_dtype=torch.bfloat16)