Update keras_hub/src/models/causal_lm.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: Mohamed-Ashraf273

keras_hub/src/models/causal_lm.py

@@ -178,7 +178,7 @@ def wrapped_generate_function(inputs, stop_token_ids=None):
                 return ov_infer(inputs, stop_token_ids, self.generate_step)
 
             self.generate_function = wrapped_generate_function
-        if keras.config.backend() == "torch":
+        elif keras.config.backend() == "torch":
             import torch
 
             def wrapped_generate_function(