Update inception_v1.py

Author: ProGamerGov

captum/optim/models/_image/inception_v1.py

@@ -266,7 +266,10 @@ def forward(
         x = self.drop(x)
         x = self.fc(x)
         if not self.aux_logits:
-            return cast(torch.Tensor, x)
+            if torch.jit.is_scripting():
+                return x
+            else
+                return cast(torch.Tensor, x)
         else:
             return x, aux1_output, aux2_output