Open amathews-amd opened 2 years ago
This seems to be a workaround for now:
diff --git a/benchmarks/xlmr/ootb/xlmr.py b/benchmarks/xlmr/ootb/xlmr.py
index b0cb790..671e998 100644
--- a/benchmarks/xlmr/ootb/xlmr.py
+++ b/benchmarks/xlmr/ootb/xlmr.py
@@ -26,7 +26,7 @@ def time_ms(use_gpu):
return time.time_ns() * 1e-6
def get_model():
- fairseq_xlmr_large = torch.hub.load('pytorch/fairseq:main', 'xlmr.large')
+ fairseq_xlmr_large = torch.hub.load('pytorch/fairseq:v0.12.0', 'xlmr.large')
# TODO use torchscript? jit/script this model?
return fairseq_xlmr_large
Yep, that should fix the versioning issue. Glad you found a fix @amathews-amd. I looked at the hub registration here https://github.com/facebookresearch/fairseq/blob/main/fairseq/models/roberta/model_xlmr.py and doesn't seem like it has changed. Not immediately sure what the problem is.
Possibly related to https://github.com/facebookresearch/fairseq/issues/4532