Skip to content

Conversation

MayDomine
Copy link
Collaborator

bmt.Block now can accept kwargs in forward function.

Description

bmt.Block can accept any kwargs variable by using inspect api.

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

How Has This Been Tested?

Now you can use model = BMTrainModelWrapper(model) to convert any huggingface models to bmtrain format.

Checklist

  • I have read the CONTRIBUTING document.
  • My code follows the code style of this project.
  • My change requires a change to the documentation.
  • I have updated the documentation accordingly.
  • I have added tests to cover my changes.
  • All new and existing tests passed.

@MayDomine MayDomine changed the title Dev bmt.Block now can accept kwargs in forward function.Dev May 13, 2025
@MayDomine MayDomine changed the title bmt.Block now can accept kwargs in forward function.Dev bmt.Block now can accept kwargs in forward function May 13, 2025
@MayDomine MayDomine requested a review from zkh2016 May 13, 2025 06:46
@MayDomine MayDomine merged commit 9140185 into main May 13, 2025
14 checks passed
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.

2 participants