fix added

SkinGPT.py

@@ -299,7 +299,7 @@ class SkinGPT4(nn.Module):
         print(f"\n[DEBUG] Before replacement:")
         print(f"Text embeddings shape: {input_embeddings.shape}")
         print(f"Visual embeddings shape: {visual_embeds.shape}")
-        print(f"Image token embedding (before):\n{input_embeddings[0, replace_positions[1], :5]}...")
+        print(f"Image token embedding (before):\n{input_embeddings[0, replace_positions, :5]}...")
 
         for pos in replace_positions:
             input_embeddings[0, pos[1]] = visual_embeds[0]
@@ -309,7 +309,7 @@ class SkinGPT4(nn.Module):
         # input_embeddings[image_token_pos] = visual_embeds
 
         print(f"\n[DEBUG] After replacement:")
-        print(f"Image token embedding (after):\n{input_embeddings[0, replace_positions[1], :5]}...")
+        print(f"Image token embedding (after):\n{input_embeddings[0, replace_positions, :5]}...")
 
         # outputs = self.llama.generate(
         #     inputs_embeds=input_embeddings,