remove static

google · May 18, 2024 · af8562f · af8562f
1 parent 2ca0d24
commit af8562f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MaxText/inference_microbenchmark.py b/MaxText/inference_microbenchmark.py
@@ -133,7 +133,7 @@ def ar_benchmark_loop(config, engine, params, decode_state, iters, profile_name)
 
 
 def ar_lowering(engine, params, decode_state):
-    lowered_generate = engine.generate.lower(params, decode_state)
+    lowered_generate = engine.generate.lower(engine, params, decode_state)
     compiled_generate = lowered_generate.compile()
     return compiled_generate