diff --git a/convert-generic.py b/convert-generic.py index 141d1fe72..addc8fdcc 100755 --- a/convert-generic.py +++ b/convert-generic.py @@ -1,11 +1,11 @@ #!/usr/bin/env python3 from __future__ import annotations -from util import parse_args import sys -import model +import model +from util import parse_args args = parse_args() diff --git a/model.py b/model.py index 94f090b2d..9ad5e46fb 100644 --- a/model.py +++ b/model.py @@ -1,15 +1,15 @@ +import contextlib +import json import os import re import sys -import json -import torch -import contextlib -import numpy as np - from enum import IntEnum from pathlib import Path from typing import TYPE_CHECKING, Any, ContextManager, Iterator, TypeAlias, cast +import numpy as np +import torch + if TYPE_CHECKING: from torch import Tensor diff --git a/util.py b/util.py index d266b54b1..3388638de 100644 --- a/util.py +++ b/util.py @@ -1,7 +1,7 @@ import argparse - from pathlib import Path + def parse_args() -> argparse.Namespace: parser = argparse.ArgumentParser(description="Convert a huggingface model to a GGML compatible file") parser.add_argument(