remove unused llm_w,llm_h

Author: shungcp

tpu_inference/models/jax/qwen2_vl.py

@@ -518,9 +518,6 @@ def __call__(self, x: jax.Array, grid_thw: tuple[tuple[int, int,
         for i in range(num_grids):
             t, h, w = grid_thw[i]
 
-            llm_h = h // self.spatial_merge_size
-            llm_w = w // self.spatial_merge_size
-
             rotary_pos_emb_thw = self.rotary_pos_emb_thw(t, h, w)
             cu_seqlens_thw = jnp.full(t, h * w, dtype=jnp.int32)
 
@@ -904,4 +901,4 @@ def load_weights(self, rng_key: jax.Array) -> None:
             for i, path in enumerate(sorted(unloaded_params)[:10]):
                 logger.error(f"  {i+1}. {path}")
 
-            raise ValueError(f"Not all parameters were loaded. Found {len(unloaded_params)} unloaded parameters.")
+            raise ValueError(f"Not all parameters were loaded. Found {len(unloaded_params)} unloaded parameters.")