Update utils.py
Browse files
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 =
|
| 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 |
)
|