huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,118 Commits 74 Branches 16 Tags
79c27fc489f2eece486fa433a0ae75c66a398e6f
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
79c27fc489 Segment-anything fixes: avoid normalizing twice. (#767) 
* Segment-anything fixes: avoid normalizing twice.

* More fixes for the image aspect ratio.
 2023-09-07 21:45:16 +01:00
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
Laurent Mazare
7b50f3e106 More segment-anything again. (#764) 
* More segment-anything again.

* Transformer block forward.

* Two-ways transformer.

* Position embeddings.

* Sketch the prompt encoder.

* More prompt-encoder.

* More prompt-encoder.

* Add the main sam module.

* Embed the transformer.

* And hook the transformer forward step.

* Build the model.

* Handle the global attn indexes.

* Get the model to load.
 2023-09-07 12:06:55 +01:00