Add model checkpoint and source

initial commit