-
Notifications
You must be signed in to change notification settings - Fork 60
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
running packages when running the "Supervised Fine-Tuning of Llama 3 8B on one AWS Trainium instance" sample #720
Comments
Do you have the names of the packages that are missing by any chance please? |
I tried to grab the neuron drivers:
and I then removed So I tried reinstall
and still got the same |
System Info
Who can help?
@michaelbenayoun @JingyaHuang
Information
Tasks
examples
folder (such as GLUE/SQuAD, ...)Reproduction (minimal, reproducible, runnable)
Precompilation step in https://huggingface.co/docs/optimum-neuron/en/training_tutorials/sft_lora_finetune_llm is failing on many missing packages. Is there a specific DLC we can use?
Expected behavior
Running the tutorial successfully. The "Fine-tune and Test Llama-3 8B on AWS Trainium" tutorial works with no issue with the same settings.
The text was updated successfully, but these errors were encountered: