From 104f88a7be64234ec58950deed8142bc7748d9da Mon Sep 17 00:00:00 2001
From: Vaatsalya <96314403+vaatsalya123@users.noreply.github.com>
Date: Thu, 22 Feb 2024 02:35:56 +0400
Subject: [PATCH] Fix demo bug.  (#75)

* Update 03_transpiling_a_tf_model.ipynb

* Update 03_transpiling_a_tf_model.ipynb

* Update 03_transpiling_a_tf_model.ipynb

* Update 03_transpiling_a_tf_model.ipynb
---
 guides/03_transpiling_a_tf_model.ipynb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/guides/03_transpiling_a_tf_model.ipynb b/guides/03_transpiling_a_tf_model.ipynb
index af9f787a..9224a02e 100644
--- a/guides/03_transpiling_a_tf_model.ipynb
+++ b/guides/03_transpiling_a_tf_model.ipynb
@@ -119,8 +119,9 @@
       "outputs": [],
       "source": [
         "# Transpile it into a torch.nn.Module with the corresponding parameters\n",
+        "ivy.set_soft_device_mode(True)\n",
         "noise = tf.random.normal(shape=(1, 224, 224, 3))\n",
-        "torch_eff_encoder = ivy.transpile(eff_encoder, to=\"torch\", args=(noise,))"
+        "torch_eff_encoder = ivy.transpile(eff_encoder, source=\"tensorflow\", to=\"torch\", args=(noise,), modes_to_trace=\"eval\")"
       ]
     },
     {