          logger.verbose("%s - %sMB free of %sMB",  # type: ignore
                         self._vram_stats["device"],
                         self._vram_stats["vram_free"],
                         self._vram_stats["vram_total"])
          if cast(int, self._vram_stats["vram_free"]) <= self._total_vram_required:
              logger.warning("Not enough free VRAM for parallel processing. "
                             "Switching to serial")
              return False
          return True
  
 
</s> remove     def _set_phases(self, multiprocess: bool) -> List[List[str]]:
 </s> add     def _set_phases(self, multiprocess: bool) -> list[list[str]]: </s> remove         if plugins_required + batch_required <= cast(int, self._vram_stats["vram_free"]):
 </s> add         if plugins_required + batch_required <= T.cast(int, self._vram_stats["vram_free"]): </s> remove         window = cast(int, self._args["avg_samples"])
 </s> add         window = T.cast(int, self._args["avg_samples"]) </s> remove         available_vram = (cast(int, self._vram_stats["vram_free"])
 </s> add         available_vram = (T.cast(int, self._vram_stats["vram_free"]) </s> remove             retval = (os.path.basename(prefix) == "envs")
 </s> add             retval = os.path.basename(prefix) == "envs"
        if T.cast(int, self._vram_stats["vram_free"]) <= self._total_vram_required:
          logger.verbose("%s - %sMB free of %sMB",  # type: ignore
                         self._vram_stats["device"],
                         self._vram_stats["vram_free"],
                         self._vram_stats["vram_total"])
        if T.cast(int, self._vram_stats["vram_free"]) <= self._total_vram_required:
              logger.warning("Not enough free VRAM for parallel processing. "
                             "Switching to serial")
              return False
          return True
  
keep keep keep keep replace keep keep keep keep keep
 Rebase code (#1326)

* Remove tensorflow_probability requirement

* setup.py - fix progress bars

* requirements.txt: Remove pre python 3.9 packages

* update apple requirements.txt

* update INSTALL.md

* Remove python<3.9 code

* setup.py - fix Windows Installer

* typing: python3.9 compliant

* Update pytest and readthedocs python versions

* typing fixes

* Python Version updates
  - Reduce max version to 3.10
  - Default to 3.10 in installers
  - Remove incompatible 3.11 tests

* Update dependencies

* Downgrade imageio dep for Windows

* typing: merge optional unions and fixes

* Updates
  - min python version 3.10
  - typing to python 3.10 spec
  - remove pre-tf2.10 code
  - Add conda tests

* train: re-enable optimizer saving

* Update dockerfiles

* Update setup.py
  - Apple Conda deps to setup.py
  - Better Cuda + dependency handling

* bugfix: Patch logging to prevent Autograph errors

* Update dockerfiles

* Setup.py - Setup.py - stdout to utf-8

* Add more OSes to github Actions

* suppress mac-os end to end test 
plugins/extract/pipeline.py