transformers
31d01150 - Refactor TFP call to just sigmoid() (#29641)

Commit

1 year ago

Refactor TFP call to just sigmoid() (#29641) * Refactor TFP call to just sigmoid() * Make sure we cast to the right dtype

References

Author

Rocketknight1

Parents