KeerthiVM commited on
Commit
dc46fc6
·
1 Parent(s): 9148d56
Files changed (1) hide show
  1. SkinGPT.py +2 -2
SkinGPT.py CHANGED
@@ -296,7 +296,7 @@ class SkinGPT4(nn.Module):
296
  print(f"\n[DEBUG] Before replacement:")
297
  print(f"Text embeddings shape: {input_embeddings.shape}")
298
  print(f"Visual embeddings shape: {visual_embeds.shape}")
299
- print(f"Image token embedding (before):\n{input_embeddings[0, replace_positions, :5]}...")
300
 
301
  for pos in replace_positions:
302
  input_embeddings[0, pos[1]] = visual_embeds[0]
@@ -306,7 +306,7 @@ class SkinGPT4(nn.Module):
306
  # input_embeddings[image_token_pos] = visual_embeds
307
 
308
  print(f"\n[DEBUG] After replacement:")
309
- print(f"Image token embedding (after):\n{input_embeddings[0, replace_positions, :5]}...")
310
 
311
  # outputs = self.llama.generate(
312
  # inputs_embeds=input_embeddings,
 
296
  print(f"\n[DEBUG] Before replacement:")
297
  print(f"Text embeddings shape: {input_embeddings.shape}")
298
  print(f"Visual embeddings shape: {visual_embeds.shape}")
299
+ print(f"Image token embedding (before):\n{input_embeddings}...")
300
 
301
  for pos in replace_positions:
302
  input_embeddings[0, pos[1]] = visual_embeds[0]
 
306
  # input_embeddings[image_token_pos] = visual_embeds
307
 
308
  print(f"\n[DEBUG] After replacement:")
309
+ print(f"Image token embedding (after):\n{input_embeddings}...")
310
 
311
  # outputs = self.llama.generate(
312
  # inputs_embeds=input_embeddings,