Update CXR_LLAVA_HF.py
Browse files- CXR_LLAVA_HF.py +1 -0
CXR_LLAVA_HF.py
CHANGED
|
@@ -295,6 +295,7 @@ class CXRLLAVAModel(PreTrainedModel):
|
|
| 295 |
cur_new_labels.append(cur_labels[image_token_start:image_token_start + 1])
|
| 296 |
cur_labels = cur_labels[image_token_start + 2:]
|
| 297 |
else:
|
|
|
|
| 298 |
cur_new_input_embeds.append(self.llama.embed_tokens(cur_input_ids[:image_token_start]))
|
| 299 |
cur_new_input_embeds.append(cur_image_features)
|
| 300 |
if labels is not None:
|
|
|
|
| 295 |
cur_new_labels.append(cur_labels[image_token_start:image_token_start + 1])
|
| 296 |
cur_labels = cur_labels[image_token_start + 2:]
|
| 297 |
else:
|
| 298 |
+
print(self.llama.device, cur_input_ids)
|
| 299 |
cur_new_input_embeds.append(self.llama.embed_tokens(cur_input_ids[:image_token_start]))
|
| 300 |
cur_new_input_embeds.append(cur_image_features)
|
| 301 |
if labels is not None:
|