This PR fixes the translation inference error in the unilm/deltalm/test.py file and sets the task attribute in the DeltaLMModel class to enable translation tasks.
Summary of Changes
Modified the from_pretrained method in the DeltaLMModel class to set the task attribute in the kwargs dictionary to 'translation' before calling the hub_utils.from_pretrained function.
Modified the model variable assignment in the unilm/deltalm/test.py file to include the task attribute in the kwargs dictionary.
These changes ensure that the task attribute is properly set in the model configuration, allowing for successful translation inference.
Please review and merge this PR. Thank you!
Fixes #1.
To checkout this PR branch, run the following command in your terminal:
To get Sweep to edit this pull request, leave a comment below or in the code. Leaving a comment in the code will only modify the file but commenting below can change the entire PR.
Description
This PR fixes the translation inference error in the
unilm/deltalm/test.pyfile and sets thetaskattribute in theDeltaLMModelclass to enable translation tasks.Summary of Changes
from_pretrainedmethod in theDeltaLMModelclass to set thetaskattribute in thekwargsdictionary to 'translation' before calling thehub_utils.from_pretrainedfunction.modelvariable assignment in theunilm/deltalm/test.pyfile to include thetaskattribute in thekwargsdictionary.These changes ensure that the
taskattribute is properly set in the model configuration, allowing for successful translation inference.Please review and merge this PR. Thank you!
Fixes #1.
To checkout this PR branch, run the following command in your terminal:
To get Sweep to edit this pull request, leave a comment below or in the code. Leaving a comment in the code will only modify the file but commenting below can change the entire PR.