Hugging Face's logo

aehrc
/
cxrmate-rrg24

Feature Extraction
Transformers
Safetensors
vision-encoder-decoder
custom_code
Model card Files
xet
 Community
4
anicolson commited on
Commit
4b0e29f
·
verified ·
1 Parent(s): 8ca1c27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -36,6 +36,8 @@ import datasets
36
  from torch.utils.data import DataLoader
37
 
38
 
 
 
39
  tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24')
40
  model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-rrg24', trust_remote_code=True)
41
  transforms = v2.Compose(
 
36
  from torch.utils.data import DataLoader
37
 
38
 
39
+ mbatch_size = 1
40
+
41
  tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24')
42
  model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-rrg24', trust_remote_code=True)
43
  transforms = v2.Compose(