Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-24 15:18:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pad: add optional padding mode

Browse source
This commit is contained in:
Cédric Deltheil 2023-10-04 17:36:04 +02:00 committed by Cédric Deltheil
parent 9b1e25e682
commit f4298f87d2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/fluxion/utils.py
View file

@ -23,8 +23,8 @@ def manual_seed(seed: int) -> None:
_manual_seed(seed) _manual_seed(seed)
def pad(x: Tensor, pad: Iterable[int], value: float = 0.0) -> Tensor: def pad(x: Tensor, pad: Iterable[int], value: float = 0.0, mode: str = "constant") -> Tensor:
return _pad(input=x, pad=pad, value=value) # type: ignore return _pad(input=x, pad=pad, value=value, mode=mode) # type: ignore
def interpolate(x: Tensor, factor: float | torch.Size, mode: str = "nearest") -> Tensor: def interpolate(x: Tensor, factor: float | torch.Size, mode: str = "nearest") -> Tensor: