wehub-resource-sync
16 lines
378 B
Python
16 lines
378 B
Python
import unittest
|
|
from modelscope import Model
|
|
|
|
from swift.utils import find_sub_module
|
|
|
|
|
|
class TestTorchUtils(unittest.TestCase):
|
|
|
|
def test_find_sub_module(self):
|
|
model = Model.from_pretrained('damo/nlp_structbert_sentence-similarity_chinese-base')
|
|
self.assertTrue(find_sub_module(model, 'query') is not None)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|