vivek-69 commited on
Commit
fc73fa6
·
verified ·
1 Parent(s): b222478

Update CXR_LLAVA_HF.py

Browse files
Files changed (1) hide show
  1. 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, add_special_tokens=False)
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