DNABERT2报错

DNABERT2报错

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:1228, in ast_to_ttir(fn, signature, specialization, c
-> 1228 generator.visit(fn.parse())
1229 except CompilationError as e:

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:1105, in CodeGenerator.visit(self, node)
1104 last_loc = self.builder.get_loc()
-> 1105 ret = super().visit(node)
1106 # Reset the location to the last one before the visit

File ~/miniconda3/envs/DNABERT/lib/python3.10/ast.py:418, in NodeVisitor.visit(self, node)
417 visitor = getattr(self, method, self.generic_visit)
--> 418 return visitor(node)

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:303, in CodeGenerator.visit_Module(self, node)
302 def visit_Module(self, node):
--> 303 ast.NodeVisitor.generic_visit(self, node)

File ~/miniconda3/envs/DNABERT/lib/python3.10/ast.py:426, in NodeVisitor.generic_visit(self, node)
425 if isinstance(item, AST):
--> 426 self.visit(item)
427 elif isinstance(value, AST):

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:1105, in CodeGenerator.visit(self, node)
1104 last_loc = self.builder.get_loc()
-> 1105 ret = super().visit(node)
1106 # Reset the location to the last one before the visit

File ~/miniconda3/envs/DNABERT/lib/python3.10/ast.py:418, in NodeVisitor.visit(self, node)
417 visitor = getattr(self, method, self.generic_visit)
--> 418 return visitor(node)

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:376, in CodeGenerator.visit_FunctionDef(self, node)
375 # visit function body
--> 376 self.visit_compound_statement(node.body)
377 # finalize function

File /opt/local/stow/pip-3.10/local/lib/python3.10/dist-packages/triton/compiler/code_generator.py:298, in CodeGenerator.visit_compound_statement(self, stmts)
297 for stmt in stmts:

解决方法
https://huggingface.co/zhihan1996/DNABERT-2-117M/discussions/23
solved problem by using triton==2.0.0.dev20221202

pip install triton==2.0.0.dev20221202

下面是一些我个人使用的版本

pip install einops==0.8.0
pip install triton==2.0.0.dev20221202
pip install einops==0.8.0 
pytorch  1.13.0+cu117（好像2.0版本的也可以但是我没用）
transformers==4.29.2  （官方推荐，但是其他版本也可以）