search

fabio-sim / Depth-Anything-ONNX

ONNX-compatible Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data
Apache License 2.0
234 stars 23 forks source link

xformers error #1

Closed xzh929 closed 1 month ago

xzh929 commented 7 months ago

RuntimeError: unsupported output type: int, from operator: xformers::efficient_attention_forward_cutlass

environment torch:2.0.1 cuda:11.7 xformers:0.0.22

when use export.py to translate vits model,it occured this error.I have changed my torch and xformers version,it dosen't work.How can I solve it?

fabio-sim commented 7 months ago

Hi @xzh929, thank you for your interest in Depth-Anything-ONNX.

The PyTorch ONNX exporter doesn't register the xformers ops, so if you would like to export the models, please uninstall xformers first.