Skip to content

Fix run_generation test commands for TRL out usage example#1621

Merged
libinta merged 1 commit into
huggingface:mainfrom
shepark:fix_trl_readme
Dec 17, 2024
Merged

Fix run_generation test commands for TRL out usage example#1621
libinta merged 1 commit into
huggingface:mainfrom
shepark:fix_trl_readme

Conversation

@shepark
Copy link
Copy Markdown
Contributor

@shepark shepark commented Dec 17, 2024

In DPO pipeline section, it shows run_generation.py command which it runs with DPO output.
The DPO output (./dpo) is generated from llama2-70b-hf, but the examples (merging/run_generation) are basing llama2-7b-hf.
It fails with wrong dimension during merging and OOM when it runs run_generation test.

@shepark shepark requested a review from regisss as a code owner December 17, 2024 04:03
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@libinta libinta added the run-test Run CI for PRs from external contributors label Dec 17, 2024
@libinta libinta merged commit bd6afbc into huggingface:main Dec 17, 2024
huijuanzh pushed a commit to huijuanzh/optimum-habana that referenced this pull request Jan 7, 2025
Liangyx2 pushed a commit to HabanaAI/optimum-habana-fork that referenced this pull request Jan 20, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

run-test Run CI for PRs from external contributors

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants