Skip to content

Commit 9c6dc67

Browse files
authored
fix: fix an error in open_clip (#748)
1 parent 8d3ce84 commit 9c6dc67

File tree

1 file changed

+2
-2
lines changed
  • examples/open_clip/src/open_clip

1 file changed

+2
-2
lines changed

examples/open_clip/src/open_clip/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,9 +147,9 @@ def __init__(
147147
self.ln_final = text.ln_final
148148
self.text_projection = text.text_projection
149149
self.attn_mask = text.attn_mask
150-
self.logit_scale = Parameter(ops.ones([]) * init_logit_scale)
150+
self.logit_scale = Parameter(ops.ones(()) * init_logit_scale)
151151
if init_logit_bias is not None:
152-
self.logit_bias = Parameter(ops.ones([]) * init_logit_bias)
152+
self.logit_bias = Parameter(ops.ones(()) * init_logit_bias)
153153
else:
154154
self.logit_bias = None
155155

0 commit comments

Comments
 (0)