Model: Format

Signed-off-by: kingbri <[email protected]>
theroyallab · Feb 18, 2024 · a19a4eb · a19a4eb
1 parent 7def32e
commit a19a4eb
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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)
             )