facebookresearch/fairseq

Request to provide torchscript-ed model

开放

#4,853 创建于 2022年11月9日

 (0 条评论) (1 个反应) (0 位负责人)Python (6,224 个派生)batch import
enhancementhelp wantedneeds triage

仓库指标

星标
 (29,107 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

🚀 Feature Request

https://huggingface.co/facebook/wav2vec2-base-960h/tree/main provides only model.state_dict(). Please also consider upload a torchscript-ed model using either torch.jit.script() or torch.jit.trace().

Motivation

It simplifies users' life as they don't need to install fairseq to use a torchscript-ed model

Additional context

We want to test models trained using CTC loss from various frameworks using k2 for decoding. (See https://github.com/k2-fsa/k2/pull/1096)

We only need a torchscript-ed model but for the current situation, we have to install fairseq and export the pretrained model by ourselves via torchscript.

贡献者指南