Skip to content

Commit ec603d7

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

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

monai/networks/blocks/text_embedding.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class TextEncoder(nn.Module):
2424
2525
Connecting text and medical 3D image, based on: "Liu et al.,
2626
CLIP-Driven Universal Model for Organ Segmentation and Tumor Detection <https://arxiv.org/pdf/2301.00785.pdf>"
27-
"""
27+
"""
2828
def __init__(
2929
self,
3030
out_channels: int,
@@ -56,6 +56,3 @@ def forward(self):
5656
# text embedding as random initialized 'rand_embedding'
5757
test_encoding = self.text_embedding.weight.unsqueeze(2).unsqueeze(2).unsqueeze(2)
5858
return test_encoding
59-
60-
61-

0 commit comments

Comments
 (0)