Fix get_dtype_name()

bigscience-workshop · Jun 30, 2023 · f8b32ec · f8b32ec
1 parent 7bf9b40
commit f8b32ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/petals/server/throughput.py b/src/petals/server/throughput.py
@@ -203,6 +203,6 @@ def get_device_name(device: torch.device) -> str:
 
 def get_dtype_name(dtype: torch.dtype, quant_type: QuantType) -> str:
     name = str(dtype)
-    if quant_type is not None:
+    if quant_type != QuantType.NONE:
         name += f", quantized to {quant_type.name.lower()}"
     return name