added input_ids to BharatAI's forward method

model.py

@@ -116,9 +116,9 @@ class BharatAI(PreTrainedModel):
         elif isinstance(module, nn.Embedding):
             torch.nn.init.normal_(module.weight, mean=0.0, std=0.02)
 
-    def forward(self, index, targets=None):
+    def forward(self, input_ids, index, targets=None):
         B, T = index.shape
-
+        x = input_ids
 
         # idx and targets are both (B,T) tensor of integers
         tok_emb = self.token_embedding_table(index) # (B,T,C)