We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8d3ce84 commit 9c6dc67Copy full SHA for 9c6dc67
examples/open_clip/src/open_clip/model.py
@@ -147,9 +147,9 @@ def __init__(
147
self.ln_final = text.ln_final
148
self.text_projection = text.text_projection
149
self.attn_mask = text.attn_mask
150
- self.logit_scale = Parameter(ops.ones([]) * init_logit_scale)
+ self.logit_scale = Parameter(ops.ones(()) * init_logit_scale)
151
if init_logit_bias is not None:
152
- self.logit_bias = Parameter(ops.ones([]) * init_logit_bias)
+ self.logit_bias = Parameter(ops.ones(()) * init_logit_bias)
153
else:
154
self.logit_bias = None
155
0 commit comments