From 8882073978583dcea0250985487db21a8029bc0e Mon Sep 17 00:00:00 2001
From: Colin <>
Date: Sun, 3 Aug 2025 18:23:26 +0800
Subject: [PATCH] Update test model.

---
 wit/inference.py          | 7 ++++---
 wit/query_block_output.py | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/wit/inference.py b/wit/inference.py
index ad242e7..9fa265e 100644
--- a/wit/inference.py
+++ b/wit/inference.py
@@ -10,7 +10,7 @@ if __name__ == "__main__":
 
     # checkpoint_path = "log/bigger/version_0/checkpoints/epoch=72-step=360328.ckpt"
     # checkpoint_path = "log/bigger/version_4/checkpoints/epoch=81-step=64288.ckpt"
-    checkpoint_path = "log/bigger/version_8/checkpoints/epoch=14-step=67455.ckpt"
+    checkpoint_path = "log/bigger/version_6/checkpoints/epoch=14-step=67455.ckpt"
 
     qwen = LightModule.load_from_checkpoint(checkpoint_path=checkpoint_path)
     qwen.eval()
@@ -29,9 +29,10 @@ if __name__ == "__main__":
     # seq:849
     # seq:991
     # seq:995
+    seq = 995
 
-    node = map.get_nodetree(995)
-    item, l, rank_idx, rank_all = map.get_sequence(995)
+    node = map.get_nodetree(seq)
+    item, l, rank_idx, rank_all = map.get_sequence(seq)
     print("len of seq:" + str(len(item)))
 
     for i in range(1, len(item)):
diff --git a/wit/query_block_output.py b/wit/query_block_output.py
index d231ec3..27157ca 100644
--- a/wit/query_block_output.py
+++ b/wit/query_block_output.py
@@ -15,7 +15,7 @@ import dataset.dataset as ds
 
 if __name__ == "__main__":
 
-    checkpoint_path = "log/bigger/version_8/checkpoints/epoch=14-step=67455.ckpt"
+    checkpoint_path = "log/bigger/version_6/checkpoints/epoch=14-step=67455.ckpt"
 
     qwen = LightModule.load_from_checkpoint(checkpoint_path=checkpoint_path)
     qwen.eval()