Update modeling_neollm.py

modeling_neollm.py

@@ -15,7 +15,7 @@ Updated to include:
 """
 
 import math
-from typing import Any, Callable, Optional, Union
+from typing import Any, Callable, Optional, Union, Tuple
 
 import torch
 import torch.nn.functional as F