                               for plugin in self._active_plugins)
          gpu_plugins = [p for p in self._current_phase if self._vram_per_phase[p] > 0]
          scaling = self._parallel_scaling.get(len(gpu_plugins), self._scaling_fallback)
          plugins_required = sum(self._vram_per_phase[p] for p in gpu_plugins) * scaling
          if plugins_required + batch_required <= cast(int, self._vram_stats["vram_free"]):
              logger.debug("Plugin requirements within threshold: (plugins_required: %sMB, "
                           "vram_free: %sMB)", plugins_required, self._vram_stats["vram_free"])
              return
          # Hacky split across plugins that use vram
          available_vram = (cast(int, self._vram_stats["vram_free"])
 
</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         phases: List[List[str]] = []
        current_phase: List[str] = []
        available = cast(int, self._vram_stats["vram_free"])
 </s> add         phases: list[list[str]] = []
        current_phase: list[str] = []
        available = T.cast(int, self._vram_stats["vram_free"]) </s> remove             x_center = (cast(int, face.left) + face.right) / 2.0
            y_center = (cast(int, face.top) + face.bottom) / 2.0 - cast(int, face.height) * 0.12
            scale = (cast(int, face.width) + cast(int, face.height)) * self.reference_scale
            center_scale[index, :, 0] = np.full(68, x_center, dtype='float32')
            center_scale[index, :, 1] = np.full(68, y_center, dtype='float32')
 </s> add             x_ctr = (T.cast(int, face.left) + face.right) / 2.0
            y_ctr = (T.cast(int, face.top) + face.bottom) / 2.0 - T.cast(int, face.height) * 0.12
            scale = (T.cast(int, face.width) + T.cast(int, face.height)) * self.reference_scale
            center_scale[index, :, 0] = np.full(68, x_ctr, dtype='float32')
            center_scale[index, :, 1] = np.full(68, y_ctr, dtype='float32') </s> remove         if cast(int, self._vram_stats["vram_free"]) <= self._total_vram_required:
 </s> add         if T.cast(int, self._vram_stats["vram_free"]) <= self._total_vram_required: </s> remove     def _set_plugin_batchsize(self, gpu_plugins: List[str], available_vram: float) -> None:
 </s> add     def _set_plugin_batchsize(self, gpu_plugins: list[str], available_vram: float) -> None:
        if plugins_required + batch_required <= T.cast(int, self._vram_stats["vram_free"]):
                               for plugin in self._active_plugins)
          gpu_plugins = [p for p in self._current_phase if self._vram_per_phase[p] > 0]
          scaling = self._parallel_scaling.get(len(gpu_plugins), self._scaling_fallback)
          plugins_required = sum(self._vram_per_phase[p] for p in gpu_plugins) * scaling
        if plugins_required + batch_required <= T.cast(int, self._vram_stats["vram_free"]):
              logger.debug("Plugin requirements within threshold: (plugins_required: %sMB, "
                           "vram_free: %sMB)", plugins_required, self._vram_stats["vram_free"])
              return
          # Hacky split across plugins that use vram
          available_vram = (cast(int, self._vram_stats["vram_free"])
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