Update inference_gohan_cid.py

inference_gohan_cid.py

@@ -58,10 +58,10 @@ class GohanCIDInferenceEngine:
 			n_num_features=8,
 			cat_cardinalities=self.cat_cardinalities,
 			d_out=len(self.all_cids),
-			d_token=768,  # Use the actual saved model's d_token
+			d_token=1024,  # Use the actual saved model's d_token
 			n_blocks=8,
 			attention_dropout=0.15,
-			ffn_d_hidden=768,  # Use the actual saved model's ffn_d_hidden
+			ffn_d_hidden=1024,  # Use the actual saved model's ffn_d_hidden
 			ffn_dropout=0.15,
 			residual_dropout=0.10
 		)