              The keyword arguments for generating time-lapse previews. If a time-lapse preview is
              not required then this should be ``None``. Otherwise all values should be full paths
              the keys being `input_a`, `input_b`, `output`.
          """
          logger.trace("Training one step: (iteration: %s)", self._model.iterations)
          do_preview = viewer is not None
          do_timelapse = timelapse_kwargs is not None
          snapshot_interval = self._model.training_opts.get("snapshot_interval", 0)
 
</s> remove             trainer.train_one_step(viewer, timelapse)
 </s> add             trainer.train_one_step(viewer, timelapse, self._args.colab) </s> remove                 self.__print_loss(self._pingpong.loss)
 </s> add                 self.__print_loss(self._pingpong.loss, is_colab) </s> remove         print("[{}] [#{:05d}] {}".format(self._timestamp,
                                         self._model.iterations,
                                         output), end='\r')
 </s> add         output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output) </s> remove                 exit(1)
 </s> add                 sys.exit(1)
        is_colab: bool
            ``True`` if FaceSwap is executing in a Google Colab session, otherwise ``False``
              The keyword arguments for generating time-lapse previews. If a time-lapse preview is
              not required then this should be ``None``. Otherwise all values should be full paths
              the keys being `input_a`, `input_b`, `output`.
        is_colab: bool
            ``True`` if FaceSwap is executing in a Google Colab session, otherwise ``False``
          """
          logger.trace("Training one step: (iteration: %s)", self._model.iterations)
          do_preview = viewer is not None
          do_timelapse = timelapse_kwargs is not None
          snapshot_interval = self._model.training_opts.get("snapshot_interval", 0)
keep keep add keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
plugins/train/trainer/_base.py