Skip to content

fix: Rollback trl to a version which doesn't overwrite labels and attention masks.#600

Closed
dushyantbehl wants to merge 1 commit intofoundation-model-stack:mainfrom
dushyantbehl:rollback-trl
Closed

fix: Rollback trl to a version which doesn't overwrite labels and attention masks.#600
dushyantbehl wants to merge 1 commit intofoundation-model-stack:mainfrom
dushyantbehl:rollback-trl

Conversation

@dushyantbehl
Copy link
Collaborator

@dushyantbehl dushyantbehl commented Aug 28, 2025

Description of the change

This PR rolbacks trl to v16.1 to solve https://github.ibm.com/ai-foundation/watson-fm-stack-tracker/issues/1913

Related issue number

How to verify the PR

Was the PR tested

  • I have added >=1 unit test(s) for every new method I have added.
  • I have ensured all unit tests pass

@github-actions
Copy link

Thanks for making a pull request! 😃
One of the maintainers will review and advise on the next steps.

@dushyantbehl dushyantbehl requested review from ashokponkumar and removed request for aluu317, anhuong, fabianlim and kmehant August 28, 2025 07:02
@github-actions github-actions bot added the fix label Aug 28, 2025
@github-actions
Copy link

Thanks for making a pull request! 😃
One of the maintainers will review and advise on the next steps.

Signed-off-by: Dushyant Behl <dushyantbehl@in.ibm.com>
@dushyantbehl dushyantbehl added the on hold This PR is on hold and will not be merged right away label Aug 28, 2025
"tokenizers>=0.13.3,<1.0",
"tqdm>=4.66.2,<5.0",
"trl>=0.13,<0.18",
"trl>=0.13,<0.17",
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If we are anyway returning the right collator, should we still revert the version?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@ashokponkumar

This is on hold for now while I figure out the workaround but it was done because the collator seq2seq has some problems with v0.17

@dushyantbehl dushyantbehl marked this pull request as draft August 28, 2025 14:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

fix on hold This PR is on hold and will not be merged right away

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments