SeanLee97 / AnglE

Train and Infer Powerful Sentence Embeddings with AnglE | 🔥 SOTA on STS and MTEB Leaderboard

Home Page:https://arxiv.org/abs/2309.12871

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Bug] Stuck before trainer.train()

mengyao00 opened this issue · comments

Hello, I cannot run examples/Angle-ATEC.ipynb, angle.fit() output nothing and GPUs are not working.
Might be version issue, my environment: Successfully installed bitsandbytes-0.41.3.post2 boltons-23.1.1 peft-0.7.1 tokenizers-0.15.0 transformers-4.36.2

I change requirements.txt to transformers==4.32.1, still not working

Successfully installed accelerate-0.25.0 bitsandbytes-0.41.3.post2 boltons-23.1.1 datasets-2.16.0 dill-0.3.7 huggingface-hub-0.20.1 multiprocess-0.70.15 peft-0.7.1 pyarrow-hotfix-0.6 safetensors-0.4.1 tokenizers-0.13.3 transformers-4.32.1 xxhash-3.4.1 works for me finally.

commented

Thanks for your debugging!