From b8cd15af715514ea3b4a7c856c336e603f187a0a Mon Sep 17 00:00:00 2001 From: aviator19941 Date: Wed, 24 Apr 2024 01:31:58 -0500 Subject: [PATCH] Fix formatting --- sharktank/sharktank/examples/validate_llama_ref_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sharktank/sharktank/examples/validate_llama_ref_model.py b/sharktank/sharktank/examples/validate_llama_ref_model.py index c8b061534..8b6a8d40c 100644 --- a/sharktank/sharktank/examples/validate_llama_ref_model.py +++ b/sharktank/sharktank/examples/validate_llama_ref_model.py @@ -15,6 +15,7 @@ def main(args: list[str]): from ..utils import cli + torch.no_grad().__enter__() parser = cli.create_parser()