mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-04-25 00:39:10 +00:00
Tree: Format
Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
@@ -402,9 +402,8 @@ class ExllamaV2Container(BaseModelContainer):
|
|||||||
self.draft_cache_mode = unwrap(draft_args.get("draft_cache_mode"), "FP16")
|
self.draft_cache_mode = unwrap(draft_args.get("draft_cache_mode"), "FP16")
|
||||||
|
|
||||||
# Catch exllamav3 draft_cache_mode
|
# Catch exllamav3 draft_cache_mode
|
||||||
if (
|
if self.draft_cache_mode != "FP16" and not self.draft_cache_mode.startswith(
|
||||||
self.draft_cache_mode != "FP16"
|
"Q"
|
||||||
and not self.draft_cache_mode.startswith("Q")
|
|
||||||
):
|
):
|
||||||
logger.warning(
|
logger.warning(
|
||||||
f"Provided draft cache mode '{self.draft_cache_mode}' is not a "
|
f"Provided draft cache mode '{self.draft_cache_mode}' is not a "
|
||||||
|
|||||||
Reference in New Issue
Block a user