Skip to content

Commit 7f3d3fc

Browse files
tangy5yuchengt
andauthored
fix unversal names (#6512)
Fix the universal names in the file. --------- Signed-off-by: yuchengt <yuchengt@nvidia.com> Co-authored-by: yuchengt <yuchengt@nvidia.com>
1 parent e5cf74a commit 7f3d3fc

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

monai/networks/blocks/text_embedding.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,9 @@
1616
from torch.utils import model_zoo
1717

1818
url_map = {
19-
"clip_encoding_univeral_model_32": (
19+
"clip_encoding_universal_model_32": (
2020
"https://github.com/Project-MONAI/MONAI-extra-test-data/"
21-
"releases/download/0.8.1/clip_encoding_univeral_model.pth"
21+
"releases/download/0.8.1/clip_encoding_universal_model.pth"
2222
)
2323
}
2424

@@ -41,7 +41,7 @@ def __init__(
4141
spatial_dims: int = 3,
4242
text_dim: int = 512,
4343
hidden_size: int = 256,
44-
encoding: str = "clip_encoding_univeral_model_32",
44+
encoding: str = "clip_encoding_universal_model_32",
4545
pretrained: bool = True,
4646
) -> None:
4747
"""

tests/test_text_encoding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ def test_test_encoding_shape(self):
2222
with skip_if_downloading_fails():
2323
# test 2D encoder
2424
text_encoder = TextEncoder(
25-
spatial_dims=2, out_channels=32, encoding="clip_encoding_univeral_model_32", pretrained=True
25+
spatial_dims=2, out_channels=32, encoding="clip_encoding_universal_model_32", pretrained=True
2626
)
2727
text_encoding = text_encoder()
2828
self.assertEqual(text_encoding.shape, (32, 256, 1, 1))
2929

3030
# test 3D encoder
3131
text_encoder = TextEncoder(
32-
spatial_dims=3, out_channels=32, encoding="clip_encoding_univeral_model_32", pretrained=True
32+
spatial_dims=3, out_channels=32, encoding="clip_encoding_universal_model_32", pretrained=True
3333
)
3434
text_encoding = text_encoder()
3535
self.assertEqual(text_encoding.shape, (32, 256, 1, 1, 1))

0 commit comments

Comments
 (0)