search

unslothai / unsloth

Finetune Llama 3, Mistral, Phi & Gemma LLMs 2-5x faster with 80% less memory
https://unsloth.ai
Apache License 2.0
12.47k stars 810 forks source link

No module named 'triton.third_party' #422

Open Yezhibin701227 opened 2 months ago

Yezhibin701227 commented 2 months ago

i install the packages of torch==2.1.2+cu118 and xformers==0.0.23.post1 in the conda, follow the steps of "Alpaca+Llama-3 8B full example.ipynb" to run on the local machine with an A6000 GPU. The error info attached with file. unsloth_train_error

Yezhibin701227 commented 2 months ago

I found out the problem , reinstalled the xformers: pip3 install -U xformers==0.0.23.post1 --index-url https://download.pytorch.org/whl/cu118

danielhanchen commented 2 months ago

Oh great you solved it!