From a19a4eb1bef9a24d5c21665753d952207ca6bab8 Mon Sep 17 00:00:00 2001 From: kingbri Date: Sun, 18 Feb 2024 18:31:31 -0500 Subject: [PATCH] Model: Format Signed-off-by: kingbri --- backends/exllamav2/model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/backends/exllamav2/model.py b/backends/exllamav2/model.py index 7608a55e..bf8b6787 100644 --- a/backends/exllamav2/model.py +++ b/backends/exllamav2/model.py @@ -114,9 +114,7 @@ def progress(loaded_modules: int, total_modules: int, # Auto GPU split parameters self.gpu_split_auto = gpu_split_auto - autosplit_reserve_megabytes = unwrap( - kwargs.get("autosplit_reserve"), [96] - ) + autosplit_reserve_megabytes = unwrap(kwargs.get("autosplit_reserve"), [96]) self.autosplit_reserve = list( map(lambda value: value * 1024**2, autosplit_reserve_megabytes) )