mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00

Files

Laurent Mazare 7396b8ed1a Segment Anything - process images (#766 )

* Start processing images.

* Add LayerNorm2d.

* Properly use LayerNorm2d.

* Tweak eps.

* Use LayerNorm on inputs with a rank different from 3.

* Window partitioning.

* Fix a couple todos.

* More todos.

* Hard-code the einsums.

* More padding support.

* Some sizes tweaks.

* Use the hub to get the weights.

* Use a batch matmul.

* Tweaks.

* More fixes.

* Get some predictions to be generated.

2023-09-07 19:22:45 +01:00