From b0005699cc0554b1f6566c1d922475c7fcab3571 Mon Sep 17 00:00:00 2001
From: randaller <randaller@users.noreply.github.com>
Date: Sun, 5 Mar 2023 20:11:38 +0300
Subject: [PATCH] Update example-cpu.py

---
 example-cpu.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/example-cpu.py b/example-cpu.py
index 3d915b9..ff37ef9 100644
--- a/example-cpu.py
+++ b/example-cpu.py
@@ -22,7 +22,7 @@ def load(
     checkpoints = sorted(Path(ckpt_dir).glob("*.pth"))
     ckpt_path = checkpoints[-1]
 
-    print("Loading models...")
+    print("Loading model...")
     checkpoint = torch.load(ckpt_path, map_location="cpu")
     with open(Path(ckpt_dir) / "params.json", "r") as f:
         params = json.loads(f.read())
@@ -40,7 +40,7 @@ def load(
     model.load_state_dict(checkpoint, strict=False)
 
     generator = LLaMA(model, tokenizer)
-    print(f"Loaded models in {time.time() - start_time:.2f} seconds")
+    print(f"Loaded model in {time.time() - start_time:.2f} seconds")
     return generator