switch to EXLA as the evaluator is lacking vectorisation support:

xhr15 · xhr15 · commit 311f77cc77db · 2025-11-14T23:26:59.000+01:00
```
** (RuntimeError) unexpected vectorized axes in evaluator for operation :add: #Nx.Tensor&lt;
       vectorized[batch: 1]
       s32[1]

       Nx.Defn.Expr
       tensor a        s32[1]
       b = reshape a   s32[1][1]
```
diff --git a/test/bumblebee/text/generation_test.exs b/test/bumblebee/text/generation_test.exs
@@ -156,7 +156,8 @@ defmodule Bumblebee.Text.GenerationTest do
     #
     # Now, with the processor below, we expect the sequence of [79, 80, 81 ..]
 
-    %{token_ids: token_ids} = generate.(params, inputs)
+    %{token_ids: token_ids} =
+      Nx.Defn.jit_apply(generate, [params, inputs], compiler: EXLA)
 
     # first token_id should be 79 as we enforce token_id 79
     assert_equal(token_ids[[0, 0]], 79)
@@ -186,7 +187,8 @@ defmodule Bumblebee.Text.GenerationTest do
         ]
       )
 
-    %{token_ids: token_ids} = generate.(params, inputs)
+    %{token_ids: token_ids} =
+      Nx.Defn.jit_apply(generate, [params, inputs], compiler: EXLA)
 
     # result without logit processor: 80, 80, 80
 

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,8 @@ defmodule Bumblebee.Text.GenerationTest do`
`156`	`156`	`#`
`157`	`157`	`# Now, with the processor below, we expect the sequence of [79, 80, 81 ..]`
`158`	`158`
`159`		`- %{token_ids: token_ids} = generate.(params, inputs)`
	`159`	`+ %{token_ids: token_ids} =`
	`160`	`+ Nx.Defn.jit_apply(generate, [params, inputs], compiler: EXLA)`
`160`	`161`
`161`	`162`	`# first token_id should be 79 as we enforce token_id 79`
`162`	`163`	`assert_equal(token_ids[[0, 0]], 79)`
`@@ -186,7 +187,8 @@ defmodule Bumblebee.Text.GenerationTest do`
`186`	`187`	`]`
`187`	`188`	`)`
`188`	`189`
`189`		`- %{token_ids: token_ids} = generate.(params, inputs)`
	`190`	`+ %{token_ids: token_ids} =`
	`191`	`+ Nx.Defn.jit_apply(generate, [params, inputs], compiler: EXLA)`
`190`	`192`
`191`	`193`	`# result without logit processor: 80, 80, 80`
`192`	`194`