Skip to content

Commit

Permalink
Update train_trainer.py
Browse files Browse the repository at this point in the history
  • Loading branch information
taishan1994 committed Jun 26, 2023
1 parent 3c98ea5 commit 11fb635
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions train_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -145,8 +145,6 @@ def main():
with open(args.deepspeed_jaon_path, "r") as fp:
deepspeed_json = json.load(fp)

import sys
sys.exit(0)

model = AutoModelForSeq2SeqLM.from_pretrained(args.model_dir,
trust_remote_code=True,
Expand Down

0 comments on commit 11fb635

Please sign in to comment.