This commit is contained in:
Yousef Rafat 2026-01-04 20:32:38 +02:00
parent f588e6c821
commit 72ca18acc2

View File

@ -478,7 +478,7 @@ class InflatedCausalConv3d(ops.Conv3d):
except NotImplementedError:
# for: Could not run 'aten::cudnn_convolution' with arguments from the 'CPU' backend
if not self.logged_once:
logging.warning("VAE is on CPU for decoding. This is most likely due to being not enough memory")
logging.warning("VAE is on CPU for decoding. This is most likely due to not enough memory")
self.logged_once = True
return F.conv3d(input, weight, bias, *args, **kwargs)