Update CXR_LLAVA_HF.py

CXR_LLAVA_HF.py

@@ -615,6 +615,9 @@ class CXRLLAVAModel(PreTrainedModel):
             input_ids = self.tokenizer_image_token(prompt, self.tokenizer, IMAGE_TOKEN_INDEX, return_tensors='pt').unsqueeze(0)
             if self.device == 'cuda':
                 input_ids = input_ids.cuda()
+                print('using cuda')
+            else:
+                print('using cpu')
             stopping_criteria = KeywordsStoppingCriteria(["</s>"], self.tokenizer, input_ids)
 
             image_args = {"images": images}