habdine commited on
Commit
2701f17
·
verified ·
1 Parent(s): 9b685b3

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -685,6 +685,7 @@ class Prot2TextTrainer(Seq2SeqTrainer):
685
  gen_kwargs["num_beams"] = (
686
  gen_kwargs["num_beams"] if gen_kwargs.get("num_beams") is not None else self.model.config.num_beams
687
  )
 
688
  gen_kwargs["synced_gpus"] = (
689
  gen_kwargs["synced_gpus"] if gen_kwargs.get("synced_gpus") is not None else default_synced_gpus
690
  )
 
685
  gen_kwargs["num_beams"] = (
686
  gen_kwargs["num_beams"] if gen_kwargs.get("num_beams") is not None else self.model.config.num_beams
687
  )
688
+ default_synced_gpus = False
689
  gen_kwargs["synced_gpus"] = (
690
  gen_kwargs["synced_gpus"] if gen_kwargs.get("synced_gpus") is not None else default_synced_gpus
691
  )