Fix run_clip.py (#20234)
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -198,7 +198,8 @@ class Transform(torch.nn.Module):
|
|||||||
Normalize(mean, std),
|
Normalize(mean, std),
|
||||||
)
|
)
|
||||||
|
|
||||||
def forward(self, x: Image) -> torch.Tensor:
|
def forward(self, x) -> torch.Tensor:
|
||||||
|
"""`x` should be an instance of `PIL.Image.Image`"""
|
||||||
with torch.no_grad():
|
with torch.no_grad():
|
||||||
x = self.transforms(x)
|
x = self.transforms(x)
|
||||||
return x
|
return x
|
||||||
|
|||||||
Reference in New Issue
Block a user