format

anakinxc · Feb 1, 2024 · 5e04c73 · 5e04c73
1 parent 90ec4ce
commit 5e04c73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/tutorials/develop_your_first_mpc_application.ipynb b/docs/tutorials/develop_your_first_mpc_application.ipynb
@@ -4202,6 +4202,7 @@
     "# import model impl which has been tested with simulator\n",
     "# from Model import model\n",
     "\n",
+    "\n",
     "def run_model(x1, x2, y):\n",
     "    # here, suppose we divide the dataset into two party\n",
     "    x = jnp.concatenate((x1, x2), axis=1)\n",

diff --git a/examples/python/ml/flax_whisper/flax_whisper.py b/examples/python/ml/flax_whisper/flax_whisper.py
@@ -73,9 +73,7 @@ def run_on_spu():
 
     input_ids = ppd.device("P1")(lambda x: x)(inputs_ids.input_features)
     params = ppd.device("P2")(lambda x: x)(pretrained_model.params)
-    outputs_ids = ppd.device("SPU")(
-        text_generation, copts = copts
-    )(input_ids, params)
+    outputs_ids = ppd.device("SPU")(text_generation, copts=copts)(input_ids, params)
     outputs_ids = ppd.get(outputs_ids)
     return outputs_ids