Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Model] Initial support for BLIP-2 #5920

Merged
merged 251 commits into from
Jul 27, 2024
Merged
Changes from 1 commit
Commits
Show all changes
251 commits
Select commit Hold shift + click to select a range
df2aa19
Move dummy data generation to input registry
DarkLight1337 Jun 3, 2024
c72d2b3
Update docs
DarkLight1337 Jun 3, 2024
d8c6488
Rename `process_input` to `map_input`
DarkLight1337 Jun 3, 2024
f18de48
Reorder arguments
DarkLight1337 Jun 3, 2024
653537d
Apply input processor
DarkLight1337 Jun 3, 2024
a2f5a3c
Remove `VisionLanguageConfig` from input mapper
DarkLight1337 Jun 3, 2024
378ad80
Fix bad use of `functools.partial`
DarkLight1337 Jun 3, 2024
7aa3778
Use default input processor
DarkLight1337 Jun 3, 2024
c774168
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 4, 2024
532f863
Fix wrong arguments
DarkLight1337 Jun 4, 2024
080d40c
Use pillow image instead of tensor to avoid bypassing the processor b…
DarkLight1337 Jun 5, 2024
662693a
Update interface of dummy data factory and input processor
DarkLight1337 Jun 5, 2024
9bc5fcc
Use `InputContext` to handle checked type cast of config types
DarkLight1337 Jun 5, 2024
911cac7
Add input processor for injecting image tokens; fix docs
DarkLight1337 Jun 5, 2024
a38b347
Add new documentation pages
DarkLight1337 Jun 5, 2024
29c3bb3
Fix LLaVA-NeXT input processor and cleanup code
DarkLight1337 Jun 5, 2024
9cfbcce
Fix LLaVA-NeXT input processor and cleanup code
DarkLight1337 Jun 5, 2024
7bb6cbf
Add sanity check
DarkLight1337 Jun 6, 2024
ccf49c4
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 6, 2024
3482d32
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 6, 2024
8ea8468
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 8, 2024
be3d64f
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 8, 2024
2ff5be6
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 10, 2024
8e2ff86
Update LLaVA-NeXT
DarkLight1337 Jun 11, 2024
553f684
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 11, 2024
b134dfc
Update name
DarkLight1337 Jun 11, 2024
1efa480
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 11, 2024
1a08444
Update LLaVA-NeXT
DarkLight1337 Jun 11, 2024
7e33706
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 11, 2024
cfc31fd
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 11, 2024
3fb622c
Remove `MULTIMODAL` convenience property as it was causing some (impo…
DarkLight1337 Jun 11, 2024
da85ab2
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 11, 2024
383bea1
Update docs
DarkLight1337 Jun 11, 2024
80a09f2
Remove double processing of image tokens
DarkLight1337 Jun 12, 2024
6a70e4f
Add docs
DarkLight1337 Jun 12, 2024
8322ecb
Add docs
DarkLight1337 Jun 12, 2024
52a0116
Add docs
DarkLight1337 Jun 12, 2024
c1733dd
Add docs
DarkLight1337 Jun 12, 2024
b7a8683
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 12, 2024
9fb5e72
Remove more instances of double processing; update docs
DarkLight1337 Jun 13, 2024
25f9949
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 13, 2024
03c7e65
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 13, 2024
3932b3f
Remove xfail
DarkLight1337 Jun 13, 2024
7fa877a
Fix missing image token in OpenAI API serving
DarkLight1337 Jun 13, 2024
092e550
Fix LLaVA-NeXT test
DarkLight1337 Jun 14, 2024
7a19862
Remove duplicate processing in async engine
DarkLight1337 Jun 14, 2024
fd7d954
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 15, 2024
49dac3e
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 15, 2024
b2c6832
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 15, 2024
0104218
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 18, 2024
18cc7e0
Set up dummy data factory for phi3v
DarkLight1337 Jun 18, 2024
2291617
Move dummy data factories to model files
DarkLight1337 Jun 18, 2024
adf5503
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 18, 2024
e5a94e4
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 18, 2024
9b0386d
Move input processors to model files
DarkLight1337 Jun 18, 2024
4e656e7
Set up input processor for phi3v
DarkLight1337 Jun 18, 2024
fecf1f0
Fix wrong feature size
DarkLight1337 Jun 18, 2024
086e0fe
Fix wrong feature size
DarkLight1337 Jun 18, 2024
8c26a18
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 19, 2024
81522fe
Fix wrong feature size
DarkLight1337 Jun 19, 2024
c036b86
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 24, 2024
f75e1ab
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 24, 2024
b24e8d9
Update validation
DarkLight1337 Jun 24, 2024
8569d35
Fix image feature calculation for phi3v
DarkLight1337 Jun 24, 2024
bfa5aa9
Remove redundant code
DarkLight1337 Jun 24, 2024
dc34121
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 24, 2024
07e695d
Apply isort
DarkLight1337 Jun 24, 2024
8a43a77
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 24, 2024
825401d
Apply yapf
DarkLight1337 Jun 24, 2024
4a0d4d1
Reduce `max_tokens` so that test still passes
DarkLight1337 Jun 25, 2024
8d22fe0
Fix vllm to hf output (+ rename)
DarkLight1337 Jun 25, 2024
2e1ee2f
Fix wrong arguments
DarkLight1337 Jun 25, 2024
7229b07
Move `DummyImageDataFactories` into CLIP model file
DarkLight1337 Jun 25, 2024
17800fd
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 25, 2024
50f994b
Move `input_processor_for_clip` into CLIP
DarkLight1337 Jun 25, 2024
838aa9b
Remove some magic numbers
DarkLight1337 Jun 25, 2024
e7a5564
Test multiscale inputs for LLaVA-NeXT
DarkLight1337 Jun 25, 2024
36e8001
Handle multiscale inputs (different number of patches per batch) in L…
DarkLight1337 Jun 25, 2024
39e6d42
Fix wrong feature size
DarkLight1337 Jun 26, 2024
0d7f18f
Apply formatter
DarkLight1337 Jun 26, 2024
8e5dc7c
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
d9a4150
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 26, 2024
6849236
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
6d02491
Revert max_tokens
DarkLight1337 Jun 26, 2024
76ddea4
Add more tests for input mapper
DarkLight1337 Jun 26, 2024
4b20e66
Sanity check: Also test multiscale inputs for LLaVA-1.5
DarkLight1337 Jun 26, 2024
784af1a
Do not auto-convert image dtype to model's dtype
DarkLight1337 Jun 26, 2024
8e5fb12
Update prompts
DarkLight1337 Jun 26, 2024
4b947ad
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 26, 2024
e7397ee
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
865be7a
Fix mapper tests w.r.t. dtype change
DarkLight1337 Jun 26, 2024
9e82a26
Clarify docs and add todo
DarkLight1337 Jun 26, 2024
46391de
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
a4733f9
Remove TODO since vision config will be removed soon
DarkLight1337 Jun 26, 2024
6b19e6c
Expand docs
DarkLight1337 Jun 26, 2024
be326f2
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
f451668
Add ref
DarkLight1337 Jun 26, 2024
5c0c8cf
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
3d7b795
Update docs
DarkLight1337 Jun 26, 2024
1abb8a7
Add docs
DarkLight1337 Jun 26, 2024
428d420
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
698830f
Fix name
DarkLight1337 Jun 26, 2024
ac9ea9a
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
334b1a9
Add `MultiModalInputs` to docs
DarkLight1337 Jun 26, 2024
36ab12d
Fix and add links
DarkLight1337 Jun 26, 2024
af01e97
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 26, 2024
c303421
Fix `is_multiscale` not provided anymore
DarkLight1337 Jun 26, 2024
0a0c0e3
Also test multiscale input for phi3v
DarkLight1337 Jun 26, 2024
60517a7
Revert max_tokens for phi3v as numerical error still persists
DarkLight1337 Jun 26, 2024
57df434
Improve error message
DarkLight1337 Jun 26, 2024
ffe0675
Log the full output for easier reference
DarkLight1337 Jun 26, 2024
c7a2a66
Update xfail to be more efficient
DarkLight1337 Jun 26, 2024
598e0e3
Also xfail llava test
DarkLight1337 Jun 26, 2024
f84d87a
Update comment
DarkLight1337 Jun 27, 2024
5dfb6fc
Update docs
DarkLight1337 Jun 27, 2024
bbeff03
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 27, 2024
bf3281c
modify llava_next
ywang96 Jun 27, 2024
56e2d3b
Update comment
DarkLight1337 Jun 27, 2024
d2f8c6d
Update docs
DarkLight1337 Jun 27, 2024
7c197d2
Use dynamic image feature size calculation
DarkLight1337 Jun 27, 2024
f5ffd3e
Fix phi3v not handling `image_sizes` correctly
DarkLight1337 Jun 27, 2024
66aad21
Apply formatter
DarkLight1337 Jun 27, 2024
d1c68c0
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 27, 2024
5f32d53
Add see also
DarkLight1337 Jun 27, 2024
15df4ef
Update examples prompt format
DarkLight1337 Jun 27, 2024
f2e4633
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 27, 2024
095e008
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 27, 2024
a6e3162
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 27, 2024
28922af
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 27, 2024
ce06541
Fix config
DarkLight1337 Jun 27, 2024
cdcc2d4
Fix config
DarkLight1337 Jun 27, 2024
4212abf
Update docs
DarkLight1337 Jun 27, 2024
07c08e3
Update docs
DarkLight1337 Jun 27, 2024
f3f5854
Fix `MultiModalInputs` not working in Python 3.8
DarkLight1337 Jun 27, 2024
bebf9e7
Fix `_ImageAssets` not working in Python 3.8
DarkLight1337 Jun 27, 2024
3f4f4bf
Merge branch 'upstream' into blip-2
DarkLight1337 Jun 27, 2024
e948275
Inital impl. BLIP-2
DarkLight1337 Jun 27, 2024
06d2339
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 27, 2024
fc83d0c
Update BLIP-2 using new API
DarkLight1337 Jun 27, 2024
eb33485
Enable test to run
DarkLight1337 Jun 27, 2024
d68c462
Fix input processor
DarkLight1337 Jun 27, 2024
b909c60
Fix wrong lm_head
DarkLight1337 Jun 27, 2024
9b85e60
Fix wrong output of vision tower
DarkLight1337 Jun 28, 2024
e864427
Fix BLIP-2 repeating output and output conversion
DarkLight1337 Jun 28, 2024
7e80ecc
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 28, 2024
487d742
Merge branch 'upstream' into mm-image-tokenizer
DarkLight1337 Jun 28, 2024
36f72b6
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 28, 2024
e354d2b
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 28, 2024
43350b8
update example
ywang96 Jun 28, 2024
57791de
update doc
ywang96 Jun 28, 2024
b2b1e11
Merge branch 'mm-image-tokenizer' into mm-image-tokenizer-2
DarkLight1337 Jun 28, 2024
5757821
Apply formatter
DarkLight1337 Jun 28, 2024
f36e099
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 28, 2024
fbc5f70
Update docs
DarkLight1337 Jun 28, 2024
dbeee10
Support `eos_token_id` from `config.json`
DarkLight1337 Jun 28, 2024
161585a
Add `trust_remote_code`
DarkLight1337 Jun 28, 2024
261dc71
Merge branch 'eos-from-config' into blip-2
DarkLight1337 Jun 28, 2024
36de6f4
Use updated EOS token detection code
DarkLight1337 Jun 28, 2024
4292ccb
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 28, 2024
6a24360
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 28, 2024
5d23a96
Apply formatter
DarkLight1337 Jun 28, 2024
9c91649
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 28, 2024
78d2e10
Apply formatter
DarkLight1337 Jun 28, 2024
78064e0
Fix OpenAI server not working for phi3v
DarkLight1337 Jun 28, 2024
4cb809c
Preemptively handle upcoming models
DarkLight1337 Jun 28, 2024
754e238
Add more models
DarkLight1337 Jun 28, 2024
9edb53c
Update feature size for dummy data
DarkLight1337 Jun 28, 2024
2795b16
Use a less strict check
DarkLight1337 Jun 29, 2024
86ffd60
Fix phi3v test
DarkLight1337 Jun 29, 2024
f339dd1
Update default length as the dummy image feature size is increased
DarkLight1337 Jun 29, 2024
59a7a4c
Raise full error if output is completely different
DarkLight1337 Jun 29, 2024
62952e1
Fix phi3v not using input processor
DarkLight1337 Jun 29, 2024
49f39d6
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 29, 2024
5810335
Update BLIP-2 test according to merged changes
DarkLight1337 Jun 29, 2024
0ce3ecb
Move size factors outside
DarkLight1337 Jun 29, 2024
379b99a
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 29, 2024
b43e8c3
Apply formatter
DarkLight1337 Jun 29, 2024
ac25521
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 29, 2024
9e2e69a
Move size factors outside
DarkLight1337 Jun 29, 2024
44dec19
Implement BLIPVisionModel in vLLM
DarkLight1337 Jun 29, 2024
9023794
Fix some outputs not being checked
DarkLight1337 Jun 29, 2024
453e144
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 29, 2024
3da4a00
Fix some outputs not being checked
DarkLight1337 Jun 29, 2024
fc5549c
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 30, 2024
f6c8061
Also test no image
DarkLight1337 Jun 30, 2024
e4f99e6
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
191c671
Also test no image
DarkLight1337 Jun 30, 2024
15cc847
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 30, 2024
235c8a9
Batch by size factors
DarkLight1337 Jun 30, 2024
b98d924
Factor out xfail code
DarkLight1337 Jun 30, 2024
a3eb4fe
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
90c60f3
Apply merge changes to BLIP-2 test
DarkLight1337 Jun 30, 2024
2c2558b
Fix unused args
DarkLight1337 Jun 30, 2024
b0d868c
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
ec28eca
Check logprobs instead of xfailing
DarkLight1337 Jun 30, 2024
5a337f5
Merge branch 'upstream' into mm-image-tokenizer-2
DarkLight1337 Jun 30, 2024
05db877
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
213e1e4
Apply merge changes to BLIP-2 tests
DarkLight1337 Jun 30, 2024
2eb3490
Fix different scales not being in the same batch
DarkLight1337 Jun 30, 2024
9b42589
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
e671249
Apply merge changes to BLIP-2 tests
DarkLight1337 Jun 30, 2024
6301a52
Apply suggestions from code review
DarkLight1337 Jun 30, 2024
14f10fc
Add link
DarkLight1337 Jun 30, 2024
7c335c3
Use `self.multi_modal_projector` directly
DarkLight1337 Jun 30, 2024
33c860e
Allow users to send image token formatted prompt directly
DarkLight1337 Jun 30, 2024
e03bc57
Factor out the code for placeholder token IDs
DarkLight1337 Jun 30, 2024
a9da5cf
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
b270ac3
Remove `-rx` flag
DarkLight1337 Jun 30, 2024
36095e1
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
3161221
Fix distributed tests
DarkLight1337 Jun 30, 2024
85d108a
Fix string mismatch warning
DarkLight1337 Jun 30, 2024
d648e32
Relax phi3v test; add TODO for llava tests
DarkLight1337 Jun 30, 2024
37fa0e7
Merge branch 'mm-image-tokenizer-2' into blip-2
DarkLight1337 Jun 30, 2024
2e4a833
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 3, 2024
d685c28
Add missing space
DarkLight1337 Jul 3, 2024
46bf209
Update test
DarkLight1337 Jul 3, 2024
74606ef
Update docs
DarkLight1337 Jul 3, 2024
04e5adc
Apply formatter
DarkLight1337 Jul 3, 2024
fc59a9e
Remove workaround - it's not needed anymore
DarkLight1337 Jul 3, 2024
8dc69b2
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 5, 2024
49b9c55
Update BLIP-2 and normalize validation/docs
DarkLight1337 Jul 5, 2024
d6fe0f7
Update docs
DarkLight1337 Jul 5, 2024
685d463
Update validation
DarkLight1337 Jul 5, 2024
b00757c
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 5, 2024
49e18b1
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 7, 2024
85c65da
Add entry to supported models
DarkLight1337 Jul 7, 2024
13324d9
Add vertical space
DarkLight1337 Jul 7, 2024
3ba9b54
Fix extra BOS
DarkLight1337 Jul 10, 2024
6d3537e
Fix extra space
DarkLight1337 Jul 10, 2024
9eca3fd
Update `output_len` calculation in HF runner
DarkLight1337 Jul 10, 2024
c7077f8
ruff
DarkLight1337 Jul 10, 2024
4c82329
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 17, 2024
da7600f
Remove "boardwalk" asset
DarkLight1337 Jul 17, 2024
1e4349c
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 23, 2024
03b27d9
Format
DarkLight1337 Jul 23, 2024
eec28e6
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 25, 2024
28cc937
yapf
DarkLight1337 Jul 25, 2024
3203ded
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 25, 2024
a24d633
Rename
DarkLight1337 Jul 25, 2024
f41b468
Fix merge
DarkLight1337 Jul 26, 2024
92047c4
Format
DarkLight1337 Jul 26, 2024
d3d1814
Fix type error
DarkLight1337 Jul 26, 2024
356ec29
Update models with recent refactors
DarkLight1337 Jul 26, 2024
3dbf3fc
Remove unused attribute
DarkLight1337 Jul 26, 2024
47d777c
Remove unused comment
DarkLight1337 Jul 26, 2024
eba25fb
Merge branch 'upstream' into blip-2
ywang96 Jul 27, 2024
cb32e09
update
ywang96 Jul 27, 2024
a0d45bb
Merge branch 'upstream' into blip-2
ywang96 Jul 27, 2024
11f9223
Add chat template for BLIP-2
DarkLight1337 Jul 27, 2024
388bae4
Merge branch 'upstream' into blip-2
DarkLight1337 Jul 27, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Remove unused comment
  • Loading branch information
DarkLight1337 committed Jul 26, 2024
commit 47d777cfb7527d311482c51deb0366168c427c74
1 change: 0 additions & 1 deletion vllm/model_executor/models/blip2.py
Original file line number Diff line number Diff line change
Expand Up @@ -371,7 +371,6 @@ def forward(
encoder_hidden_states=encoder_hidden_states,
query_length=query_length,
)
# pooled_output = sequence_output[:, 0, :]

return sequence_output

Expand Down
Loading