fix: configuration types

huggingface · Feb 23, 2025 · 7696204 · 7696204
1 parent 08003a5
commit 7696204
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/transformers/models/efficientloftr/configuration_efficientloftr.py b/src/transformers/models/efficientloftr/configuration_efficientloftr.py
@@ -110,8 +110,8 @@ def __init__(
         self,
         stage_block_dims: List[int] = None,
         stage_num_blocks: List[int] = None,
-        stage_hidden_expansion: List[int | float] = None,
-        stage_stride: List[int | float] = None,
+        stage_hidden_expansion: List[float] = None,
+        stage_stride: List[int] = None,
         hidden_size: int = 256,
         activation_function: str = "relu",
         aggregation_sizes: List[int] = None,

diff --git a/src/transformers/models/efficientloftr/modular_efficientloftr.py b/src/transformers/models/efficientloftr/modular_efficientloftr.py
@@ -265,8 +265,8 @@ def __init__(
         self,
         stage_block_dims: List[int] = None,
         stage_num_blocks: List[int] = None,
-        stage_hidden_expansion: List[int | float] = None,
-        stage_stride: List[int | float] = None,
+        stage_hidden_expansion: List[float] = None,
+        stage_stride: List[int] = None,
         hidden_size: int = 256,
         activation_function: str = "relu",
         aggregation_sizes: List[int] = None,