generation.py

Author: lucylq

examples/models/llama2/runner/generation.py

@@ -1,10 +1,9 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 
 import argparse
-
-import json
 from typing import List, Optional, Tuple, TypedDict
 
+import json
 import torch
 import torch.nn.functional as F
 from executorch.examples.models.llama2.llama_transformer import ModelArgs