[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

llama_inference_speed.py

@@ -3,9 +3,9 @@
 import torch
 from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
 from torch.distributed.fsdp.wrap import transformer_auto_wrap_policy
-
-from transformers import LlamaTokenizer, LlamaForCausalLM
+from transformers import LlamaForCausalLM, LlamaTokenizer
 from transformers.models.llama.modeling_llama import LlamaDecoderLayer
+
 # from https://github.com/huggingface/transformers/issues/22687
 # torchrun --nproc_per_node=2 --master_port=56718 run_forward.py
 model_dir = "huggyllama/llama-13b"