fix added
Browse files- 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
|
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
|
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,
|