Replace scripts and model with improved version

Update scripts to work around collator valueerror. Update weights

Saving weights and logs of step 300

Add config, tokenizer and training script