I’m not sure how the config.json and args.bin files are used, but you would need to use the model source code in order to create an instance of it before loading the state_dict.
How are you creating an object of the model at the moment?