mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-09-11 17:25:31 +03:00
a54021305d
Summary: # Before submitting - [ ] Was this discussed/approved via a Github issue? (no need for typos, doc improvements) - [ ] Did you read the [contributor guideline](https://github.com/pytorch/fairseq/blob/main/CONTRIBUTING.md)? - [ ] Did you make sure to update the docs? - [ ] Did you write any new necessary tests? ## What does this PR do? fix `black` failures ## PR review Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in Github issues there's a high chance it will not be merged. ## Did you have fun? Make sure you had fun coding � Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/2816 Reviewed By: alexeib Differential Revision: D33172615 Pulled By: dianaml0 fbshipit-source-id: 36b141f42941670f1bfa981041d878042feb0428
30 lines
796 B
Python
30 lines
796 B
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
import unittest
|
|
|
|
import torch
|
|
|
|
try:
|
|
import huggingface_hub
|
|
except ImportError:
|
|
huggingface_hub = None
|
|
|
|
from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
|
|
|
|
|
|
@unittest.skipIf(not huggingface_hub, "Requires huggingface_hub install")
|
|
class TestHuggingFaceHub(unittest.TestCase):
|
|
@torch.no_grad()
|
|
def test_hf_fastspeech2(self):
|
|
hf_model_id = "facebook/fastspeech2-en-ljspeech"
|
|
models, cfg, task = load_model_ensemble_and_task_from_hf_hub(hf_model_id)
|
|
self.assertTrue(len(models) > 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|