use torch.jit.Final instead of Final for beit, eva

Author: rwightman

timm/models/beit.py

@@ -39,8 +39,7 @@
 # --------------------------------------------------------'
 
 import math
-from functools import partial
-from typing import Callable, Final, Optional, Tuple, Union
+from typing import Callable, Optional, Tuple, Union
 
 import torch
 import torch.nn as nn
@@ -80,7 +79,7 @@ def gen_relative_position_index(window_size: Tuple[int, int]) -> torch.Tensor:
 
 
 class Attention(nn.Module):
-    fused_attn: Final[bool]
+    fused_attn: torch.jit.Final[bool]
 
     def __init__(
             self,

timm/models/eva.py

@@ -26,7 +26,7 @@
 # EVA02 models Copyright (c) 2023 BAAI-Vision
 
 import math
-from typing import Callable, Final, Optional, Tuple, Union
+from typing import Callable, Optional, Tuple, Union
 
 import torch
 import torch.nn as nn
@@ -45,7 +45,7 @@
 
 
 class EvaAttention(nn.Module):
-    fused_attn: Final[bool]
+    fused_attn: torch.jit.Final[bool]
 
     def __init__(
             self,