chore: fix precommit (#51)

.pre-commit-config.yaml

@@ -17,7 +17,7 @@ repos:
       hooks:
           - id: black
             language_version: python3
-    - repo: https://gitlab.com/pycqa/flake8
-      rev: 3.9.2
+    - repo: https://github.com/PyCQA/flake8
+      rev: 6.0.0
       hooks:
           - id: flake8

manifest/api/models/huggingface.py

@@ -3,7 +3,6 @@ import json
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple, Union, cast
 
-import deepspeed
 import numpy as np
 import PIL
 import torch
@@ -25,6 +24,7 @@ from transformers import (
     PreTrainedTokenizer,
 )
 
+import deepspeed
 from manifest.api.models.model import Model
 
 MODEL_REGISTRY = {