facebookresearch/fairseq

Request to provide torchscript-ed model

オープン

#4,853 opened on 2022/11/09

 (0 件のコメント) (1 件のリアクション) (0 人の担当者)Python (6,224 件のフォーク)batch import
enhancementhelp wantedneeds triage

Repository metrics

Stars
 (29,107 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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

コントリビューターガイド