4AI / LS-LLaMA

A Simple but Powerful SOTA NER Model | Official Code For Label Supervised LLaMA Finetuning
https://arxiv.org/abs/2310.01208
MIT License
131 stars 20 forks source link

Assertion failed #13

Open catalwaysright opened 6 months ago

catalwaysright commented 6 months ago

I met this ../aten/src/ATen/native/cuda/Indexing.cu:1292: indexSelectLargeIndex: block: [10,0,0], thread: [31,0,0] AssertionsrcIndex < srcSelectDimSizefailed when trying out other 13b llama models. Any clue?

BillYao-6 commented 4 months ago

I have the same problems,didyou solve it ?

SeanLee97 commented 4 months ago

Could you provide more details? It is hard to locate the problem without enough context.

Which backbone do you use? If the backbone is not LLaMA, it is better to specify the target_modules for Lora config explicitly.