Skip to content

Commit 9f8277e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 2b584e0 commit 9f8277e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tiatoolbox/models/architecture/vanilla.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
import timm
99
import torch
1010
import torchvision.models as torch_models
11-
from torch import nn
1211
from timm.layers import SwiGLUPacked
12+
from torch import nn
1313

1414
from tiatoolbox.models.models_abc import ModelABC
1515

@@ -164,7 +164,7 @@ def _get_timm_architecture(
164164
"hf_hub:paige-ai/Virchow",
165165
pretrained=pretrained,
166166
mlp_layer=SwiGLUPacked,
167-
act_layer=torch.nn.SiLU
167+
act_layer=torch.nn.SiLU,
168168
)
169169

170170
if arch_name == "Virchow2": # pragma: no cover
@@ -174,7 +174,7 @@ def _get_timm_architecture(
174174
"hf_hub:paige-ai/Virchow2",
175175
pretrained=pretrained,
176176
mlp_layer=SwiGLUPacked,
177-
act_layer=torch.nn.SiLU
177+
act_layer=torch.nn.SiLU,
178178
)
179179

180180
msg = f"Backbone {arch_name} not supported. "

0 commit comments

Comments
 (0)