Update CXR_LLAVA_HF.py
Browse files- 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
|
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}
|