enhancementhelp wantedneeds triage
倉庫指標
- Star
- (29,107 star)
- PR 合併指標
- (30 天內沒有已合併 PR)
描述
Hello.
Is there any plans to support torch.export? It would be awesome for developers who wants compilation of fairseq for backends or their deep learning accelerators.
The code would be like below.
from fairseq.models import FairseqEncoderDecoderModel
import torch
translator= FairseqEncoderDecoderModel.from_pretrained(
# ..
)
example_inputs = translator.dummy_inputs
with torch.no_grad():
ep = torch.export.export(translator_model, example_inputs)