Spaces:
Paused
Paused
Increase default live decode budget
Browse files- demo_controller.py +2 -2
- engines/live_request.py +2 -0
- scripts/space_dense_runner.py +2 -2
- scripts/space_dotcache_runner.py +6 -2
demo_controller.py
CHANGED
|
@@ -11,7 +11,7 @@ from engines.compare import build_summary_sentence, compute_agreement, compute_m
|
|
| 11 |
from engines.dense_runner import run_dense_live
|
| 12 |
from engines.dotcache_runner import run_dotcache_live
|
| 13 |
from engines.fixture_builder import FIXTURE_VERSION, build_fixture_result
|
| 14 |
-
from engines.live_request import resolve_live_runtime_settings
|
| 15 |
from engines.presets import (
|
| 16 |
BIT_WIDTHS,
|
| 17 |
CONTEXT_LENGTHS,
|
|
@@ -250,7 +250,7 @@ def _execute_live_request(request: DemoRequest, cache: ResultCache) -> Controlle
|
|
| 250 |
def _validate_live_request(request_payload: dict[str, Any]) -> None:
|
| 251 |
resolve_live_runtime_settings(
|
| 252 |
request_payload,
|
| 253 |
-
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS",
|
| 254 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 255 |
)
|
| 256 |
|
|
|
|
| 11 |
from engines.dense_runner import run_dense_live
|
| 12 |
from engines.dotcache_runner import run_dotcache_live
|
| 13 |
from engines.fixture_builder import FIXTURE_VERSION, build_fixture_result
|
| 14 |
+
from engines.live_request import DEFAULT_LIVE_DECODE_STEPS, resolve_live_runtime_settings
|
| 15 |
from engines.presets import (
|
| 16 |
BIT_WIDTHS,
|
| 17 |
CONTEXT_LENGTHS,
|
|
|
|
| 250 |
def _validate_live_request(request_payload: dict[str, Any]) -> None:
|
| 251 |
resolve_live_runtime_settings(
|
| 252 |
request_payload,
|
| 253 |
+
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
|
| 254 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 255 |
)
|
| 256 |
|
engines/live_request.py
CHANGED
|
@@ -14,6 +14,8 @@ SHORTLIST_TOP_K = {
|
|
| 14 |
"aggressive": 2,
|
| 15 |
}
|
| 16 |
|
|
|
|
|
|
|
| 17 |
SELECTIVE_EXACT_K_LAYERS = (19, 23)
|
| 18 |
|
| 19 |
|
|
|
|
| 14 |
"aggressive": 2,
|
| 15 |
}
|
| 16 |
|
| 17 |
+
DEFAULT_LIVE_DECODE_STEPS = 48
|
| 18 |
+
|
| 19 |
SELECTIVE_EXACT_K_LAYERS = (19, 23)
|
| 20 |
|
| 21 |
|
scripts/space_dense_runner.py
CHANGED
|
@@ -10,7 +10,7 @@ if str(REPO_ROOT) not in sys.path:
|
|
| 10 |
sys.path.insert(0, str(REPO_ROOT))
|
| 11 |
|
| 12 |
from dotcache.integrations.qwen35 import Qwen35TextHarness # noqa: E402
|
| 13 |
-
from engines.live_request import resolve_live_runtime_settings # noqa: E402
|
| 14 |
from scripts.space_runner_common import ( # noqa: E402
|
| 15 |
configure_model_cache_env,
|
| 16 |
decode_generated_text,
|
|
@@ -51,7 +51,7 @@ def main() -> int:
|
|
| 51 |
request = load_request_from_stdin()
|
| 52 |
settings = resolve_live_runtime_settings(
|
| 53 |
request,
|
| 54 |
-
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS",
|
| 55 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 56 |
)
|
| 57 |
|
|
|
|
| 10 |
sys.path.insert(0, str(REPO_ROOT))
|
| 11 |
|
| 12 |
from dotcache.integrations.qwen35 import Qwen35TextHarness # noqa: E402
|
| 13 |
+
from engines.live_request import DEFAULT_LIVE_DECODE_STEPS, resolve_live_runtime_settings # noqa: E402
|
| 14 |
from scripts.space_runner_common import ( # noqa: E402
|
| 15 |
configure_model_cache_env,
|
| 16 |
decode_generated_text,
|
|
|
|
| 51 |
request = load_request_from_stdin()
|
| 52 |
settings = resolve_live_runtime_settings(
|
| 53 |
request,
|
| 54 |
+
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
|
| 55 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 56 |
)
|
| 57 |
|
scripts/space_dotcache_runner.py
CHANGED
|
@@ -11,7 +11,11 @@ if str(REPO_ROOT) not in sys.path:
|
|
| 11 |
|
| 12 |
from benchmarks.bench_qwen35_attention_subset_dotcache_serving import _build_dotcache_config # noqa: E402
|
| 13 |
from dotcache.integrations.qwen35 import Qwen35AttentionSubsetDotCacheHarness # noqa: E402
|
| 14 |
-
from engines.live_request import
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
from scripts.space_runner_common import ( # noqa: E402
|
| 16 |
configure_model_cache_env,
|
| 17 |
decode_generated_text,
|
|
@@ -149,7 +153,7 @@ def main() -> int:
|
|
| 149 |
request = load_request_from_stdin()
|
| 150 |
settings = resolve_live_runtime_settings(
|
| 151 |
request,
|
| 152 |
-
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS",
|
| 153 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 154 |
)
|
| 155 |
|
|
|
|
| 11 |
|
| 12 |
from benchmarks.bench_qwen35_attention_subset_dotcache_serving import _build_dotcache_config # noqa: E402
|
| 13 |
from dotcache.integrations.qwen35 import Qwen35AttentionSubsetDotCacheHarness # noqa: E402
|
| 14 |
+
from engines.live_request import ( # noqa: E402
|
| 15 |
+
DEFAULT_LIVE_DECODE_STEPS,
|
| 16 |
+
resolve_live_runtime_settings,
|
| 17 |
+
selective_exact_k_overrides,
|
| 18 |
+
)
|
| 19 |
from scripts.space_runner_common import ( # noqa: E402
|
| 20 |
configure_model_cache_env,
|
| 21 |
decode_generated_text,
|
|
|
|
| 153 |
request = load_request_from_stdin()
|
| 154 |
settings = resolve_live_runtime_settings(
|
| 155 |
request,
|
| 156 |
+
decode_steps=int(os.getenv("DOTCACHE_SPACE_LIVE_DECODE_STEPS", str(DEFAULT_LIVE_DECODE_STEPS))),
|
| 157 |
max_live_context=int(os.getenv("DOTCACHE_SPACE_MAX_LIVE_CONTEXT", "4096")),
|
| 158 |
)
|
| 159 |
|