[Helios] Remove lru_cache for better AoTI compatibility and cleaner code (huggingface#13282)

SHYuanBest · web-flow · commit 0c01a4b5e2fc · 2026-03-18T23:41:58.000+05:30
fix: drop lru_cache for better AoTI compatibility
diff --git a/src/diffusers/models/transformers/transformer_helios.py b/src/diffusers/models/transformers/transformer_helios.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import math
-from functools import lru_cache
 from typing import Any
 
 import torch
@@ -343,7 +342,6 @@ def get_frequency_batched(self, freqs_base, pos):
         return freqs.cos(), freqs.sin()
 
     @torch.no_grad()
-    @lru_cache(maxsize=32)
     def _get_spatial_meshgrid(self, height, width, device_str):
         device = torch.device(device_str)
         grid_y_coords = torch.arange(height, device=device, dtype=torch.float32)
