NiuTrans.NMT icon indicating copy to clipboard operation
NiuTrans.NMT copied to clipboard

Published 20 hours ago • verified publisher icon NiuTrans

ModelConverter.py not working

Open bolon667 opened this issue 7 months ago • 0 comments

Quick and dirty fix Comment this string

#print(model['encoder.history.weight'])

Add argument weights_only=False to torch.load

state = torch.load(args.i, map_location='cpu', weights_only=False)

bolon667 avatar Mar 13 '25 19:03 bolon667