fix added

SkinGPT.py

@@ -251,7 +251,8 @@ class SkinGPT4(nn.Module):
         prompt = """### Skin Diagnosis Analysis ###
         <IMAGE>
         Could you describe the skin condition in this image?
-        Please provide a detailed analysis including possible diagnoses."""
+        Please provide a detailed analysis including possible diagnoses.
+        ### Response:"""
 
 
         print(f"\n[DEBUG] Raw Prompt:\n{prompt}")
@@ -296,14 +297,14 @@ 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}")
-        replaced_pos = replace_positions[0][0]
-        print(f"Image token embedding (before):\n{input_embeddings[0, replaced_pos, :5]}...")
+        print(f"Image token at {replace_positions[0][1].item()}:")
+        print(f"Image token embedding (before):\n{input_embeddings[0, replace_positions[0][1], :5]}...")
 
         for pos in replace_positions:
             input_embeddings[0, pos[1]] = visual_embeds[0]
 
         print(f"\n[DEBUG] After replacement:")
-        print(f"Image token embedding (after):\n{input_embeddings[0, replaced_pos, :5]}...")
+        print(f"Image token embedding (after):\n{input_embeddings[0, replace_positions[0][1], :5]}...")
 
         outputs = self.llama.generate(
             inputs_embeds=input_embeddings,