Commit
·
7042a21
1
Parent(s):
2f9ec5a
Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,7 @@ After, refer the path to OFA-base to `ckpt_dir`, and prepare an image for the te
|
|
| 44 |
>>> patch_img = patch_resize_transform(img).unsqueeze(0)
|
| 45 |
|
| 46 |
|
| 47 |
-
|
| 48 |
>>> model = OFAModel.from_pretrained(ckpt_dir, use_cache=True)
|
| 49 |
>>> generator = sequence_generator.SequenceGenerator(
|
| 50 |
tokenizer=tokenizer,
|
|
@@ -58,7 +58,7 @@ After, refer the path to OFA-base to `ckpt_dir`, and prepare an image for the te
|
|
| 58 |
>>> gen_output = generator.generate([model], data)
|
| 59 |
>>> gen = [gen_output[i][0]["tokens"] for i in range(len(gen_output))]
|
| 60 |
|
| 61 |
-
|
| 62 |
>>> model = OFAModel.from_pretrained(ckpt_dir, use_cache=False)
|
| 63 |
>>> gen = model.generate(inputs, patch_images=patch_img, num_beams=5, no_repeat_ngram_size=3)
|
| 64 |
|
|
|
|
| 44 |
>>> patch_img = patch_resize_transform(img).unsqueeze(0)
|
| 45 |
|
| 46 |
|
| 47 |
+
# using the generator of fairseq version
|
| 48 |
>>> model = OFAModel.from_pretrained(ckpt_dir, use_cache=True)
|
| 49 |
>>> generator = sequence_generator.SequenceGenerator(
|
| 50 |
tokenizer=tokenizer,
|
|
|
|
| 58 |
>>> gen_output = generator.generate([model], data)
|
| 59 |
>>> gen = [gen_output[i][0]["tokens"] for i in range(len(gen_output))]
|
| 60 |
|
| 61 |
+
# using the generator of huggingface version
|
| 62 |
>>> model = OFAModel.from_pretrained(ckpt_dir, use_cache=False)
|
| 63 |
>>> gen = model.generate(inputs, patch_images=patch_img, num_beams=5, no_repeat_ngram_size=3)
|
| 64 |
|