68109fcc80b02c45cd069c2d220b8b9e0705ac3d,plugins/extract/pipeline.py,Extractor,_set_plugin_batchsize,#Any#Any#,463

Before Change

    def _set_plugin_batchsize(plugin, available_vram):
         Set the batch size for the given plugin based on given available vram 
        plugin.batchsize = int(max(1, available_vram // plugin.vram_per_batch))
        logger.verbose("Reset batchsize for %s to %s", plugin.name, plugin.batchsize)

    def _join_threads(self):
         Join threads for current pass 
        for plugin in self._active_plugins:

After Change

        Do not update plugins which have a vram_per_batch of 0 (CPU plugins) due to
        zero division error.
        
        if plugin.vram_per_batch != 0:
            plugin.batchsize = int(max(1, available_vram // plugin.vram_per_batch))
            logger.verbose("Reset batchsize for %s to %s", plugin.name, plugin.batchsize)

    def _join_threads(self):
         Join threads for current pass 
        for plugin in self._active_plugins:
            plugin.join()

Instances