This commit is contained in:
@@ -0,0 +1,15 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user