1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

fix: Fixed type annotations for compatability with python 3.8 (#7648)

* Fixed type annotations for compatability with python 3.8

* Add required imports.
This commit is contained in:
Sai-Suraj-27
2024-04-19 11:04:09 +05:30
committed by GitHub
parent 3cfe187dc7
commit db969cc16d
3 changed files with 13 additions and 13 deletions

View File

@@ -637,7 +637,7 @@ def _filter2d(input, kernel):
height, width = tmp_kernel.shape[-2:]
padding_shape: list[int] = _compute_padding([height, width])
padding_shape: List[int] = _compute_padding([height, width])
input = torch.nn.functional.pad(input, padding_shape, mode="reflect")
# kernel and input tensor reshape to align element-wise or batch-wise params