colin
/
Witllm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix model path.

This commit is contained in:
Colin 2025-03-18 15:58:08 +08:00
parent 7faf629d45
commit cda7f04e49
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wit/inference.py
View File

@ -1,7 +1,7 @@
import torch
from wit.model.light_module import LightModule
from wit.model.light_module import ModelRunner
from model.light_module import LightModule
from model.light_module import ModelRunner
import numpy as np
import dataset.dataset as ds
@ -12,7 +12,7 @@ if __name__ == "__main__":
checkpoint_path = "log/bigger/version_1/checkpoints/epoch=14-step=74040.ckpt"
checkpoint_path = "log/bigger/version_3/checkpoints/epoch=46-step=231992.ckpt"
checkpoint_path = "log/bigger/version_8/checkpoints/epoch=49-step=246800.ckpt"
# checkpoint_path = "log/bigger/version_11/checkpoints/epoch=25-step=128336.ckpt"
# checkpoint_path = "log/bigger/version_6/checkpoints/epoch=43-step=217184.ckpt"
qwen = LightModule.load_from_checkpoint(checkpoint_path=checkpoint_path)
qwen.eval()

4
wit/query_block_output.py
View File

@ -1,7 +1,7 @@
import torch
from wit.model.light_module import LightModule
from wit.model.light_module import ModelRunner
from model.light_module import LightModule
from model.light_module import ModelRunner
import numpy as np
import math

2
wit/query_meaning_freq.py
View File

@ -1,7 +1,7 @@
import pytorch_lightning as pl
import torch
from wit.model.light_module import LightModule
from model.light_module import LightModule
from model.modeling_wit import ModelRunner
from model.tokenization_qwen import QWenTokenizer
import numpy as np