Update CXR_LLAVA_HF.py

CXR_LLAVA_HF.py

@@ -617,7 +617,7 @@ class CXRLLAVAModel(PreTrainedModel):
                 input_ids = input_ids.cuda()
                 print('using cuda')
             else:
-                print('using cpu')
+                print(f'using device {self.device}')
             stopping_criteria = KeywordsStoppingCriteria(["</s>"], self.tokenizer, input_ids)
 
             image_args = {"images": images}