Skip to content

Commit f6d0f6a

Browse files
Fix for missing enum in hpu_model_runner.py
1 parent 565245f commit f6d0f6a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/worker/hpu_model_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import os
1616
import time
1717
from array import array
18-
from enum import IntEnum
18+
from enum import Enum, IntEnum
1919
from typing import (TYPE_CHECKING, Any, Callable, Dict, List, NamedTuple,
2020
Optional, Set, Tuple, Type, TypeVar, Union)
2121

0 commit comments

Comments
 (0)