Fix error of make style

Signed-off-by: yuanwu <[email protected]>
huggingface · Sep 14, 2024 · eea1811 · eea1811
1 parent 285ee0a
commit eea1811
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/optimum/gptq/quantizer.py b/optimum/gptq/quantizer.py
@@ -546,9 +546,7 @@ def tmp(_, input, output):
 
         if self.bits == 4:
             # device not on gpu
-            if device.type != "cuda" or (
-                has_device_map and any(d in devices for d in ["cpu", "disk", "hpu"])
-            ):
+            if device.type != "cuda" or (has_device_map and any(d in devices for d in ["cpu", "disk", "hpu"])):
                 if not self.disable_exllama:
                     logger.warning(
                         "Found modules on cpu/disk. Using Exllama/Exllamav2 backend requires all the modules to be on GPU. Setting `disable_exllama=True`"