From 36eac86d6c50a24069777251cec1bd32c1271b06 Mon Sep 17 00:00:00 2001 From: erichan1 <30481032+erichan1@users.noreply.github.com> Date: Thu, 7 Oct 2021 12:16:44 -0700 Subject: [PATCH] Update README to match master->main branch change (#3922) Summary: Loading xlmr doesn't work because trying to pull from master branch, which has been changed to main. Pull Request resolved: https://github.com/pytorch/fairseq/pull/3922 Reviewed By: erichan1 Differential Revision: D31415771 Pulled By: dianaml0 fbshipit-source-id: a4b1b533b5596a4d3556c97c2f15847e444aeea9 --- examples/xlmr/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/xlmr/README.md b/examples/xlmr/README.md index b95bfe15d3..bba7910e30 100644 --- a/examples/xlmr/README.md +++ b/examples/xlmr/README.md @@ -71,7 +71,7 @@ Model | average | en | es | de | ar | hi | vi | zh ##### Load XLM-R from torch.hub (PyTorch >= 1.1): ```python import torch -xlmr = torch.hub.load('pytorch/fairseq', 'xlmr.large') +xlmr = torch.hub.load('pytorch/fairseq:main', 'xlmr.large') xlmr.eval() # disable dropout (or leave in train mode to finetune) ```