From 1cc15b2b9798374e22463a7c2a856a8eb8158db4 Mon Sep 17 00:00:00 2001
From: randaller <randaller@users.noreply.github.com>
Date: Sun, 5 Mar 2023 18:24:19 +0300
Subject: [PATCH] Update example-cpu.py

---
 example-cpu.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/example-cpu.py b/example-cpu.py
index a71e718..7311308 100644
--- a/example-cpu.py
+++ b/example-cpu.py
@@ -11,6 +11,7 @@ import json
 from pathlib import Path
 from llama import ModelArgs, Transformer, Tokenizer, LLaMA
 
+
 def load(
         ckpt_dir: str,
         tokenizer_path: str,
@@ -42,6 +43,7 @@ def load(
     print(f"Loaded models in {time.time() - start_time:.2f} seconds")
     return generator
 
+
 def main(
         ckpt_dir: str = './model',
         tokenizer_path: str = './tokenizer/tokenizer.model',
@@ -94,5 +96,6 @@ def main(
 
     print(f"Inference took {time.time() - start_time:.2f} seconds")
 
+
 if __name__ == "__main__":
     fire.Fire(main)