FlagEmbedding icon indicating copy to clipboard operation
FlagEmbedding copied to clipboard

pass kwargs into AutoModel.from_pretrained

Open ayang opened this issue 11 months ago • 1 comments

比如local_files_only必须传进去,要不断网后没法使用。直接把所有kwargs传进去是没问题的,因为 transformers/models/auto/auto_factory.py from_pretrained 函数对参数进行了过滤。

ayang avatar Jan 25 '25 03:01 ayang

你好,@ayang,谢谢提交这个 PR。考虑到简洁性,local_files_only 这个参数是不是可以用本地的 model_name_or_path 绝对路径替代?或者通过指定 cache_dir 参数来解决呢?

hanhainebula avatar Feb 07 '25 07:02 hanhainebula