Update modeling_avey.py

modeling_avey.py

@@ -12,6 +12,8 @@ from .configuration_avey import AveyConfig
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from torch.utils.checkpoint import checkpoint
 
+torch._dynamo.config.allow_unspec_int_on_nn_module = True
+
 class Contextualizer(nn.Module):
     def __init__(self, config: AveyConfig, layer_idx):
         super().__init__()