facebookresearch/fairseq

Request to provide torchscript-ed model

Open

#4,853 建立於 2022年11月9日

在 GitHub 查看
 (0 留言) (1 反應) (0 負責人)Python (6,224 fork)batch import
enhancementhelp wantedneeds triage

倉庫指標

Star
 (29,107 star)
PR 合併指標
 (30 天內沒有已合併 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.

貢獻者指南