Update inference.py

src/itwinai/torch/inference.py

@@ -154,7 +154,7 @@ def distribute_model(self) -> None:
                 "self.model is None! Mandatory constructor argument "
             )
         distribute_kwargs = {}
-        # Distributed model, optimizer, and scheduler
+        # Distributed model
         self.model, _, _ = self.strategy.distributed(
             self.model, None, None, **distribute_kwargs
         )