search

RUC-GSAI / YuLan-Rec

MIT License
287 stars 19 forks source link

Pretrained embedding data not found #20

Closed nicksukie closed 2 months ago

nicksukie commented 2 months ago

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.

Paitesanshi commented 2 months ago

@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.