cleanup

gptqmodel/utils/model.py

@@ -3,15 +3,16 @@
 import json
 import logging
 import os
+import shutil
 from logging import getLogger
 from typing import List, Optional
-from huggingface_hub import HfApi, hf_hub_download
-import shutil
+
 import accelerate
 import threadpoolctl as tctl
 import torch
 import torch.nn as nn
 import transformers
+from huggingface_hub import HfApi, hf_hub_download
 from tqdm import tqdm
 from transformers import AutoConfig, PretrainedConfig
 from transformers.utils.hub import cached_file
@@ -648,4 +649,4 @@ def copy_py_files(save_dir, file_extension=".py", model_id_or_path=""):
         for file in model_info.siblings:
             if file.rfilename.endswith(file_extension):
                 _ = hf_hub_download(repo_id=model_id_or_path, filename=file.rfilename,
-                                                  local_dir=save_dir)
+                                                  local_dir=save_dir)