jcsagar commited on
Commit
f6bb5db
·
verified ·
1 Parent(s): 30ac77c

Update CXR_LLAVA_HF.py

Browse files
Files changed (1) hide show
  1. CXR_LLAVA_HF.py +1 -1
CXR_LLAVA_HF.py CHANGED
@@ -617,7 +617,7 @@ class CXRLLAVAModel(PreTrainedModel):
617
  input_ids = input_ids.cuda()
618
  print('using cuda')
619
  else:
620
- print('using cpu')
621
  stopping_criteria = KeywordsStoppingCriteria(["</s>"], self.tokenizer, input_ids)
622
 
623
  image_args = {"images": images}
 
617
  input_ids = input_ids.cuda()
618
  print('using cuda')
619
  else:
620
+ print(f'using device {self.device}')
621
  stopping_criteria = KeywordsStoppingCriteria(["</s>"], self.tokenizer, input_ids)
622
 
623
  image_args = {"images": images}