mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-03-14 15:57:27 +00:00
OAI: Add cache_mode parameter to model
Mistakenly forgot that the user can choose what cache mode to use when loading a model. Also add when fetching model info. Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
1
main.py
1
main.py
@@ -82,6 +82,7 @@ async def get_current_model():
|
||||
rope_scale = model_container.config.scale_pos_emb,
|
||||
rope_alpha = model_container.config.scale_alpha_value,
|
||||
max_seq_len = model_container.config.max_seq_len,
|
||||
cache_mode = "FP8" if model_container.cache_fp8 else "FP16",
|
||||
prompt_template = unwrap(model_container.prompt_template, "auto")
|
||||
),
|
||||
logging = gen_logging.config
|
||||
|
||||
Reference in New Issue
Block a user