"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a138d71ec11f2f9c35f98ddae1aebb7f681a1f0b"
Unverified Commit 8265469b authored by ProGamerGov's avatar ProGamerGov Committed by GitHub
Browse files

Correct make_grid's pad_value variable type hint (#5182)

parent 3e79d149
...@@ -19,7 +19,7 @@ def make_grid( ...@@ -19,7 +19,7 @@ def make_grid(
normalize: bool = False, normalize: bool = False,
value_range: Optional[Tuple[int, int]] = None, value_range: Optional[Tuple[int, int]] = None,
scale_each: bool = False, scale_each: bool = False,
pad_value: int = 0, pad_value: float = 0.0,
**kwargs, **kwargs,
) -> torch.Tensor: ) -> torch.Tensor:
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment