linting change

nutanix · Jan 4, 2024 · bcef8ed · bcef8ed
1 parent a36e022
commit bcef8ed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/llm/kubeflow_inference_run.py b/llm/kubeflow_inference_run.py
@@ -388,8 +388,10 @@ def execute(params: argparse.Namespace) -> None:
  model_info["repo_version"] = check_if_valid_version(model_info, mount_path)
 
  if quantize_bits and int(quantize_bits) not in [4, 8]:
- print("## Quantization precision bits should be either 4 or 8."
- " Default precision used is 16 (bfloat16)")
+ print(
+ "## Quantization precision bits should be either 4 or 8."
+ " Default precision used is 16 (bfloat16)"
+ )
  sys.exit(1)
  elif quantize_bits and deployment_resources["gpus"]:
  print("## BitsAndBytes Quantization requires GPUs")