hf_text-generation-inference/server/text_generation_server/utils/speculate.py

12 lines
173 B
Python

SPECULATE = None
def get_speculate() -> int:
global SPECULATE
return SPECULATE
def set_speculate(speculate: int):
global SPECULATE
SPECULATE = speculate