TMElyralab

/

lyraXVERSE

like 2

我在a100 80G上运行失败，报了这个错误，我是按照文档步骤在镜像环境下一步一步运行的，文件分布也是按照官方上传的结构，我不知道错误的原因在哪（可恶），希望大佬能够帮帮忙，不胜感激
Traceback (most recent call last):
File "demo.py", line 13, in
model = lyraXVERSE(model_path,
File "/workspace/lyra_xverse/lyra_xverse.py", line 25, in init
self.model, self.tokenizer = self.load_model_and_tokenizer()
File "/workspace/lyra_xverse/lyra_xverse.py", line 128, in load_model_and_tokenizer
model = XVERSEModel(**model_args)
File "/workspace/lyra_xverse/model.py", line 93, in init
self.model = torch.classes.FasterTransformer.LlamaOp(
RuntimeError: Wrong Size for bin file load