ECOFRI
/

CXR-LLAVA-v2

Feature Extraction

Model card Files Files and versions

Update CXR_LLAVA_HF.py

#4

by vivek-69 - opened Apr 13, 2025

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

CXR_LLAVA_HF.py +1 -1

CXR_LLAVA_HF.py CHANGED Viewed

@@ -46,7 +46,7 @@ class CXRLLAVAModel(PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
-        self.tokenizer = transformers.LlamaTokenizer.from_pretrained(config._name_or_path, add_special_tokens=False)
         self.tokenizer.pad_token = self.tokenizer.unk_token
         self.tokenizer.sep_token = self.tokenizer.unk_token
         self.tokenizer.cls_token = self.tokenizer.unk_token

     def __init__(self, config):
         super().__init__(config)
+        self.tokenizer = transformers.LlamaTokenizer.from_pretrained(config._name_or_path)
         self.tokenizer.pad_token = self.tokenizer.unk_token
         self.tokenizer.sep_token = self.tokenizer.unk_token
         self.tokenizer.cls_token = self.tokenizer.unk_token