TRM-coding
/

PythonCopilot

Text Generation

text-generation-inference

Model card Files Files and versions

TRM-coding commited on Jun 8, 2024

Commit

63c2af3

·

verified ·

1 Parent(s): f9cc63a

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -142,7 +142,7 @@ logger.info(accelerator.state)
 # Load model and tokenizer
 if accelerator.is_main_process:
     hf_repo = Repository("./", clone_from=project_name, revision=run_name,git_email='22301131@bjtu.edu.cn')
-model = GPT2LMHeadModel.from_pretrained("TRM-coding/codeparrot-small")
 model.gradient_checkpointing=True
 tokenizer = AutoTokenizer.from_pretrained("TRM-coding/Origin-vocab")

 # Load model and tokenizer
 if accelerator.is_main_process:
     hf_repo = Repository("./", clone_from=project_name, revision=run_name,git_email='22301131@bjtu.edu.cn')
+model = GPT2LMHeadModel.from_pretrained("TRM-coding/Model-origin-small")
 model.gradient_checkpointing=True
 tokenizer = AutoTokenizer.from_pretrained("TRM-coding/Origin-vocab")