File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
tiatoolbox/models/architecture Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 88import timm
99import torch
1010import torchvision .models as torch_models
11- from torch import nn
1211from timm .layers import SwiGLUPacked
12+ from torch import nn
1313
1414from 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. "
You can’t perform that action at this time.
0 commit comments