fix: fix an error in open_clip (#748)

mindspore-lab · Nov 25, 2023 · 9c6dc67 · 9c6dc67
1 parent 8d3ce84
commit 9c6dc67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/open_clip/src/open_clip/model.py b/examples/open_clip/src/open_clip/model.py
@@ -147,9 +147,9 @@ def __init__(
         self.ln_final = text.ln_final
         self.text_projection = text.text_projection
         self.attn_mask = text.attn_mask
-        self.logit_scale = Parameter(ops.ones([]) * init_logit_scale)
+        self.logit_scale = Parameter(ops.ones(()) * init_logit_scale)
         if init_logit_bias is not None:
-            self.logit_bias = Parameter(ops.ones([]) * init_logit_bias)
+            self.logit_bias = Parameter(ops.ones(()) * init_logit_bias)
         else:
             self.logit_bias = None