Skip to content

Blip2 float32 ITM#1

Merged
NielsRogge merged 1 commit intoNielsRogge:blip2_float32from
jpizarrom:blip2_float32_itm
May 1, 2024
Merged

Blip2 float32 ITM#1
NielsRogge merged 1 commit intoNielsRogge:blip2_float32from
jpizarrom:blip2_float32_itm

Conversation

@jpizarrom
Copy link

@jpizarrom jpizarrom commented May 1, 2024

This PR removes autocast for ITM model, needed for huggingface/transformers#29261

@NielsRogge
Copy link
Owner

Thanks!

@NielsRogge NielsRogge merged commit 44e0c2e into NielsRogge:blip2_float32 May 1, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants