1580158
/
sd-webui-inpaint-anything
mirror of https://github.com/Uminosachi/sd-webui-inpaint-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove torch.jit.script from SAM2Transforms class

main
Uminosachi 2024-08-06 19:02:54 +09:00
parent e54e285072
commit 247d388ad2
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sam2/utils/transforms.py
View File

@ -25,11 +25,9 @@ class SAM2Transforms(nn.Module):
self.mean = [0.485, 0.456, 0.406] self.mean = [0.485, 0.456, 0.406]
self.std = [0.229, 0.224, 0.225] self.std = [0.229, 0.224, 0.225]
self.to_tensor = ToTensor() self.to_tensor = ToTensor()
self.transforms = torch.jit.script( self.transforms = nn.Sequential(
nn.Sequential( Resize((self.resolution, self.resolution), antialias=True),
Resize((self.resolution, self.resolution), antialias=True), Normalize(self.mean, self.std),
Normalize(self.mean, self.std),
)
) )
def __call__(self, x): def __call__(self, x):