Update CXR_LLAVA_HF.py
Browse files- CXR_LLAVA_HF.py +1 -1
CXR_LLAVA_HF.py
CHANGED
@@ -46,7 +46,7 @@ class CXRLLAVAModel(PreTrainedModel):
|
|
46 |
def __init__(self, config):
|
47 |
super().__init__(config)
|
48 |
|
49 |
-
self.tokenizer = transformers.LlamaTokenizer.from_pretrained(config._name_or_path
|
50 |
self.tokenizer.pad_token = self.tokenizer.unk_token
|
51 |
self.tokenizer.sep_token = self.tokenizer.unk_token
|
52 |
self.tokenizer.cls_token = self.tokenizer.unk_token
|
|
|
46 |
def __init__(self, config):
|
47 |
super().__init__(config)
|
48 |
|
49 |
+
self.tokenizer = transformers.LlamaTokenizer.from_pretrained(config._name_or_path)
|
50 |
self.tokenizer.pad_token = self.tokenizer.unk_token
|
51 |
self.tokenizer.sep_token = self.tokenizer.unk_token
|
52 |
self.tokenizer.cls_token = self.tokenizer.unk_token
|