bb3e3d0f1633b66023dd0a0b82b41f69381af6ff,plugins/extract/pipeline.py,Extractor,_total_vram_required,#Extractor#,256

Before Change

    @property
    def _total_vram_required(self):
         Return vram required for all phases plus the buffer 
        vrams = [getattr(self, "_{}".format(p)).vram for p in self._flow]
        vram_required_count = sum(1 for p in vrams if p > 0)
        retval = (sum(vrams) * self._parallel_scaling[vram_required_count]) + self._vram_buffer
        logger.trace(retval)
        return retval

After Change

    @property
    def _total_vram_required(self):
         Return vram required for all phases plus the buffer 
        vrams = dict()
        for phase in self._flow:
            plugin_type, idx = self._get_plugin_type_and_index(phase)
            attr = getattr(self, "_{}".format(plugin_type))
            attr = attr[idx] if idx is not None else attr
            vrams[phase] = attr.vram
        vram_required_count = sum(1 for p in vrams.values() if p > 0)
        logger.debug("VRAM requirements: %s. Plugins requiring VRAM: %s",
                     vrams, vram_required_count)
        retval = (sum(vrams.values()) *

Instances