Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CustomMBartDecoder does not support an attention implementation through torch.nn.functional.scaled_dot_product_attention #1564

Closed
Thirty502 opened this issue Jan 17, 2025 · 2 comments
Labels
bug Something isn't working

Comments

@Thirty502
Copy link

Description of the bug | 错误描述

CustomMBartDecoder does not support an attention implementation through torch.nn.functional.scaled_dot_product_attention

How to reproduce the bug | 如何复现

使用命令行进行测试

Operating system | 操作系统

Linux

Python version | Python 版本

3.10

Software version | 软件版本 (magic-pdf --version)

1.0.x

Device mode | 设备模式

cuda

@Thirty502 Thirty502 added the bug Something isn't working label Jan 17, 2025
@Xlt3h
Copy link

Xlt3h commented Jan 22, 2025

try new venv, it solved my problem. the issue might be different version of libraries :).

@myhloli
Copy link
Collaborator

myhloli commented Jan 22, 2025

#999

@myhloli myhloli closed this as completed Jan 22, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

3 participants