remove model. prefix from state_dict keys

update model file to only contain state_dict[state][model] values

add model, tokenizer, and src code