Update modeling_phi.py

modeling_phi.py

@@ -928,27 +928,74 @@ class PhiModel(PhiPreTrainedModel):
         return hidden_states
 
 group_definitions = [
-    list(range(0, 4)),  # Indices for the first group
-    list(range(4, 8)), # Indices for the second group, with overlap
-    list(range(8, 12)), # Indices for the second group, with overlap
-    list(range(12, 16)), # Indices for the second group, with overlap
-    list(range(16, 20)), # Indices for the second group, with overlap
-    list(range(20, 24)), # Indices for the second group, with overlap
-    list(range(24, 28)), # Indices for the second group, with overlap
-    list(range(28, 32)), # Indices for the second group, with overlap
-    # Add more groups as needed
+    list(range(0, 1)),  # Indices for the 1st group
+    list(range(1, 2)),
+    list(range(2, 3)),
+    list(range(3, 4)),
+    list(range(4, 5)),
+    list(range(5, 6)),
+    list(range(6, 7)),
+    list(range(7, 8)),
+    list(range(8, 9)),
+    list(range(9, 10)),
+    list(range(10, 11)),
+    list(range(11, 12)),
+    list(range(12, 13)),
+    list(range(13, 14)),
+    list(range(14, 15)),
+    list(range(15, 16)),
+    list(range(16, 17)),
+    list(range(17, 18)),
+    list(range(18, 19)),
+    list(range(19, 20)),
+    list(range(20, 21)),
+    list(range(21, 22)),
+    list(range(22, 23)),
+    list(range(23, 24)),
+    list(range(24, 25)),
+    list(range(25, 26)),
+    list(range(26, 27)),
+    list(range(27, 28)),
+    list(range(28, 29)),
+    list(range(29, 30)),
+    list(range(30, 31)),
+    list(range(31, 32)),
 ]
 
 repetitions = [
-    1, # Repetitions for the first group
-    2, # Repetitions for the second group
-    2, # Repetitions for the second group
-    2, # Repetitions for the second group
-    2, # Repetitions for the second group
-    2, # Repetitions for the second group
-    2, # Repetitions for the second group
-    1, # Repetitions for the second group
-    # Add more repetition counts as needed
+    1,
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    2, 
+    1, 
+
 ]
 
 class AdvancedSharedLayerModule(nn.Module):