habdine commited on
Commit
a450806
·
verified ·
1 Parent(s): 1341ae9

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -685,7 +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
- default_synced_gpus = True if is_deepspeed_zero3_enabled() else 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
  )
 
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
  )