pycharm调试transformers(hugging face)的模型

使用pycharm,远程调试服务器上的代码时,transformer的代码跳转不进去,会失败。可以尝试如下方案:
1、找到路径/home/usr/.cache/huggingface/modules/transformers_modules/model_name/,其中usr和model_name按自己的配置进行替换,model_name为加载的huggingface模型
2、把路径下的py文件复制到自己的代码的文件夹下
3、将原来的huggingface的模型定义代码替换为刚刚复制的代码中的模型,例如将AutoModelForCausalLM.from_pretrained替换为your_model.from_pretrainedAutoProcessor.from_pretrained替换为your_processor.from_pretrained其中your_modelyour_processor在第2步里复制的py文件里找
4、单步调试时就可以进入模型的内置函数了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值