Skip to content

Add dia#37941

Closed
ArthurZucker wants to merge 70 commits into
mainfrom
add-dia
Closed

Add dia#37941
ArthurZucker wants to merge 70 commits into
mainfrom
add-dia

Conversation

@ArthurZucker

@ArthurZucker ArthurZucker commented May 3, 2025

Copy link
Copy Markdown
Collaborator

What does this PR do?

Superseeds #37832 as it was on the main branch did not want to push to it!

Todos:

  • cleanup cache, attention masks and cache positions.
  • cleanup generation, might be a way to make it generate compatible? if you always sample the top_k *num_channel tokens, you are getting the correct number of tokens.

Comment thread src/transformers/models/dia/modeling_dia.py Outdated
@ArthurZucker ArthurZucker mentioned this pull request May 26, 2025
@vasqu

vasqu commented Jun 27, 2025

Copy link
Copy Markdown
Collaborator

Closing as #38405 supersedes this :)

@vasqu vasqu closed this Jun 27, 2025
@vasqu vasqu deleted the add-dia branch July 11, 2025 11:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants