Files
sglang/python/sglang/srt/speculative/spec_info.py
2024-12-29 00:45:57 -08:00

20 lines
352 B
Python

from enum import IntEnum, auto
class SpeculativeAlgorithm(IntEnum):
EAGLE = auto()
def is_eagle(self):
return self == SpeculativeAlgorithm.EAGLE
@staticmethod
def from_string(name: str):
name_map = {
"EAGLE": SpeculativeAlgorithm.EAGLE,
}
return name_map[name]
class SpecInfo:
pass