test 1

ROCm · Oct 9, 2024 · 4cb8d4c · 4cb8d4c
1 parent 1d2314b
commit 4cb8d4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -1773,8 +1773,8 @@ def _wrap_model(self, model, training=True, dataloader=None):
             return smp.DistributedModel(model, backward_passes_per_step=self.args.gradient_accumulation_steps)
 
         # train/eval could be run multiple-times - if already wrapped, don't re-wrap it again
-        if self.accelerator.unwrap_model(model) is not model:
-        #if unwrap_model(model) is not model:
+        #if self.accelerator.unwrap_model(model) is not model:
+        if unwrap_model(model) is not model:
             if self.args.ort:
                 from torch_ort import ORTModule
                 if type(model) is not ORTModule: