DNABERT_2 icon indicating copy to clipboard operation
DNABERT_2 copied to clipboard

finetune error

Open maris205 opened this issue 1 year ago • 3 comments

         ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/opt/conda/lib/python3.11/site-packages/triton/compiler/compiler.py", line 113, in make_ir return ast_to_ttir(self.fn, self, context=context, options=options, codegen_fns=codegen_fns) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ triton.compiler.errors.CompilationError: at 114:14: else: if EVEN_HEADDIM: k = tl.load(k_ptrs + start_n * stride_kn, mask=(start_n + offs_n)[:, None] < seqlen_k, other=0.0) else: k = tl.load(k_ptrs + start_n * stride_kn, mask=((start_n + offs_n)[:, None] < seqlen_k) & (offs_d[None, :] < headdim), other=0.0) qk = tl.zeros([BLOCK_M, BLOCK_N], dtype=tl.float32) qk += tl.dot(q, k, trans_b=True)

maris205 avatar Oct 29 '24 10:10 maris205

pip uninstall triton will run

triton seems no use for finetune in small dataset.......

maris205 avatar Oct 29 '24 12:10 maris205

pip uninstall triton will run

triton seems no use for finetune in small dataset.......

It's useful.!!!!

WhiteHe001 avatar Jan 17 '25 08:01 WhiteHe001

I was able to install a compatible version of triton by using the legacy-backend release from their github, as I describe here.

AdamInTokyo avatar Jan 29 '25 08:01 AdamInTokyo