Skip to content

fix rope dtype from torch.float64 to torch.float32#13456

Open
HsiaWinter wants to merge 23 commits intohuggingface:mainfrom
HsiaWinter:add-ernie-image
Open

fix rope dtype from torch.float64 to torch.float32#13456
HsiaWinter wants to merge 23 commits intohuggingface:mainfrom
HsiaWinter:add-ernie-image

Conversation

@HsiaWinter
Copy link
Copy Markdown
Contributor

What does this PR do?

Fixes # (issue)

change rope dtype from torch.float64 to torch.float32 to support more gpu backends.

@github-actions github-actions bot added documentation Improvements or additions to documentation models tests utils pipelines size/L PR with diff > 200 LOC labels Apr 13, 2026
@github-actions github-actions bot added size/L PR with diff > 200 LOC and removed size/L PR with diff > 200 LOC labels Apr 13, 2026
@yiyixuxu
Copy link
Copy Markdown
Collaborator

can you rebase and resolve the conflicts? the entire ernie image model & pipeline are showing up as changes here in the PR

Fix attention_mask broadcasting for NPU compatibility
@github-actions github-actions bot added size/L PR with diff > 200 LOC and removed size/L PR with diff > 200 LOC labels Apr 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation models pipelines size/L PR with diff > 200 LOC tests utils

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants