fix lint error

pytorch · Jun 16, 2023 · e7559e7 · e7559e7
1 parent 64685cb
commit e7559e7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/examples/large_models/inferentia2/save_split_checkpoints.py b/examples/large_models/inferentia2/save_split_checkpoints.py
@@ -2,8 +2,8 @@
 import os
 
 import torch
-from transformers_neuronx.module import save_pretrained_split
 from transformers.models.opt import OPTForCausalLM
+from transformers_neuronx.module import save_pretrained_split
 
 
 def create_directory_if_not_exists(path_str: str) -> str:
@@ -51,5 +51,3 @@ def amp_callback(model: OPTForCausalLM, dtype: torch.dtype) -> None:
 save_pretrained_split(hf_model, args.save_path)
 
 print(f"Files for '{args.model_name}' have been downloaded to '{args.save_path}'.")
-
-