vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use 'IntEnum' instead of 'Enum'

Browse source
This commit is contained in:
Galunid 2023-10-31 22:23:26 +01:00
parent f4b9a7ea02
commit 3ec89dcc69
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
model.py
View file

@ -7,14 +7,14 @@ import torch
import contextlib import contextlib
import numpy as np import numpy as np
from enum import Enum from enum import IntEnum
from pathlib import Path from pathlib import Path
from typing import TypeAlias, Any, Generator from typing import TypeAlias, Any, Generator
NDArray: TypeAlias = 'np.ndarray[Any, Any]' NDArray: TypeAlias = 'np.ndarray[Any, Any]'
class SentencePieceTokenTypes(Enum): class SentencePieceTokenTypes(IntEnum):
NORMAL = 1 NORMAL = 1
UNKNOWN = 2 UNKNOWN = 2
CONTROL = 3 CONTROL = 3