Skip to content

Commit 0c01a4b

Browse files
authored
[Helios] Remove lru_cache for better AoTI compatibility and cleaner code (#13282)
fix: drop lru_cache for better AoTI compatibility
1 parent 8e4b560 commit 0c01a4b

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

src/diffusers/models/transformers/transformer_helios.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
# limitations under the License.
1414

1515
import math
16-
from functools import lru_cache
1716
from typing import Any
1817

1918
import torch
@@ -343,7 +342,6 @@ def get_frequency_batched(self, freqs_base, pos):
343342
return freqs.cos(), freqs.sin()
344343

345344
@torch.no_grad()
346-
@lru_cache(maxsize=32)
347345
def _get_spatial_meshgrid(self, height, width, device_str):
348346
device = torch.device(device_str)
349347
grid_y_coords = torch.arange(height, device=device, dtype=torch.float32)

0 commit comments

Comments
 (0)