From b16ea47cd9d6a916ca88d48c844f1d162e8dd138 Mon Sep 17 00:00:00 2001 From: Mergen Nachin Date: Wed, 14 Feb 2024 10:56:44 -0800 Subject: [PATCH] Fix lint Summary: I wanna make this dashboard all green: https://hud.pytorch.org/hud/pytorch/executorch/main/1?per_page=50 this is the lint error: https://github.com/pytorch/executorch/actions/runs/7895524148/job/21548003385 Reviewed By: lucylq Differential Revision: D53763587 fbshipit-source-id: 810eba37136dc957151975993017449334a957ba --- examples/models/llama2/export_llama_lib.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py index 885850988d..268b68cc2e 100644 --- a/examples/models/llama2/export_llama_lib.py +++ b/examples/models/llama2/export_llama_lib.py @@ -352,11 +352,11 @@ def _export_llama(modelname, args) -> str: # noqa: C901 if ( args.dtype_override == "fp16" and metadata["get_dtype"] != 5 ) or args.quantization_mode == "int4": - print(f"model.to torch.float16") + print("model.to torch.float16") model = model.to(dtype=torch.float16) metadata["get_dtype"] = 5 elif args.dtype_override == "fp32" and metadata["get_dtype"] != 6: - print(f"model.to torch.float32") + print("model.to torch.float32") model = model.to(dtype=torch.float32) metadata["get_dtype"] = 6 else: