Closed xyxxmb closed 7 months ago
When I set cache_latents is True, it seems that the shape is not match.
The error is "RuntimeError: Given groups=1, weight of size [320, 5, 3, 3], expected input[4, 17, 512, 512] to have 5 channels, but got 17 channels instead"
The cache_latents function is not implemented now.
When I set cache_latents is True, it seems that the shape is not match.