Skip to content
代码片段 群组 项目
提交 2865f913 编辑于 作者: comfyanonymous's avatar comfyanonymous
浏览文件

Free memory before doing tiled decode.

上级 b49616f9
No related branches found
No related tags found
无相关合并请求
......@@ -362,7 +362,8 @@ class VAE:
return pixel_samples
def decode_tiled(self, samples, tile_x=None, tile_y=None, overlap=None):
model_management.load_model_gpu(self.patcher)
memory_used = self.memory_used_decode(samples.shape, self.vae_dtype) #TODO: calculate mem required for tile
model_management.load_models_gpu([self.patcher], memory_required=memory_used)
dims = samples.ndim - 2
args = {}
if tile_x is not None:
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册