Fix TypeError in non_uniform_interpolation

jshuadvd · Jun 14, 2024 · 01925bb · 01925bb
1 parent def4bf1
commit 01925bb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main.py b/src/main.py
@@ -39,6 +39,10 @@ def non_uniform_interpolation(pos_embed, extension_ratio, lambda_factors, n_hat)
     Returns:
         torch.Tensor: Interpolated position embeddings.
     """
+
+    if extension_ratio is None:
+        raise ValueError("extension_ratio cannot be None")
+
     d_model = pos_embed.shape[-1]
     interpolated_pos = pos_embed.clone()