aehrc
/

cxrmate-rrg24

Feature Extraction

vision-encoder-decoder

Model card Files Files and versions

anicolson commited on 10 days ago

Commit

8a12800

·

verified ·

1 Parent(s): ef90c27

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -39,8 +39,8 @@ from torch.utils.data import DataLoader
 mbatch_size = 1
 device = 'cuda'
-tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24').to(device=device)
-model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-rrg24', trust_remote_code=True)
 transforms = v2.Compose(
     [
         v2.PILToTensor(),

 mbatch_size = 1
 device = 'cuda'
+tokenizer = transformers.AutoTokenizer.from_pretrained('aehrc/cxrmate-rrg24')
+model = transformers.AutoModel.from_pretrained('aehrc/cxrmate-rrg24', trust_remote_code=True).to(device=device)
 transforms = v2.Compose(
     [
         v2.PILToTensor(),