RUC-GSAI/YuLan-Rec

Pretrained embedding data not found

nicksukie opened this issue · 1 comments

Hi,

I am trying to use RecAgent on my own dataset. I am not sure what to input for model_name under get_embedding_model() in utils.py.

With the current model as /data/pretrain_dir/all-mpnet-base-v2", I am receiving the error ValueError: Path /data/pretrain_dir/all-mpnet-base-v2 not found.

Thanks for your assistance.

Regards.

@nicksukie Hi,

You just need to change the path in utils/utils.py at line 420 to sentence-transformers/all-mpnet-base-v2, and then you can download and use the model directly from Hugging Face. This should resolve the issue. Let me know if you need further assistance!

Regards.