mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-26 22:30:19 +08:00
.
This commit is contained in:
parent
f588e6c821
commit
72ca18acc2
@ -478,7 +478,7 @@ class InflatedCausalConv3d(ops.Conv3d):
|
|||||||
except NotImplementedError:
|
except NotImplementedError:
|
||||||
# for: Could not run 'aten::cudnn_convolution' with arguments from the 'CPU' backend
|
# for: Could not run 'aten::cudnn_convolution' with arguments from the 'CPU' backend
|
||||||
if not self.logged_once:
|
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
|
self.logged_once = True
|
||||||
return F.conv3d(input, weight, bias, *args, **kwargs)
|
return F.conv3d(input, weight, bias, *args, **kwargs)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user