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.

貢獻者指南