transformers
816f4424 - Only cast logits to float when computing loss (#34147)

Commit

1 year ago

Only cast logits to float when computing loss (#34147) * Only cast logits to float when computing loss Some misses from #31292 and #33902 * Move logits.float() into existing if labels is not None branch

References

#62 - Add initial DEIMv2 model implementation

#58 - Add EoMT DINOv3 model

#59 - Fix attention mask handling in EoMT-DINOv3 converter

#29969 - [SigLIP] Add fast tokenizer

#41212 - Add EoMT with DINOv3 backbone

#39821 - Support MetaCLIP 2

#34147 - Only cast logits to float when computing loss

Author

Matthew Hoffman

Parents

e46e3bc1

transformers 816f4424 - Only cast logits to float when computing loss (#34147)

transformers
816f4424 - Only cast logits to float when computing loss (#34147)