MCPcopy
hub / github.com/modelscope/ms-swift / test_bert

Function test_bert

tests/export/test_quant.py:44–49  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

42
43
44def test_bert():
45 from swift import ExportArguments, export_main
46 output_dir = 'output/swift_test_bert_merged'
47 export_main(ExportArguments(adapters='swift/test_bert', merge_lora=True, output_dir=output_dir))
48 export_main(
49 ExportArguments(model=output_dir, load_data_args=True, quant_bits=4, quant_method='gptq', max_length=512))
50
51
52def test_reward_model():

Callers

nothing calls this directly

Calls 2

export_mainFunction · 0.90
ExportArgumentsClass · 0.90

Tested by

no test coverage detected