summaryrefslogtreecommitdiff
path: root/candle-examples/examples/segment-anything/model_transformer.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-09-08 09:39:10 +0100
committerGitHub <noreply@github.com>2023-09-08 09:39:10 +0100
commitc1453f00b11c9dd12c5aa81fb4355ce47d22d477 (patch)
tree70fac003f636d26db3b1df5ef56980500700c0f1 /candle-examples/examples/segment-anything/model_transformer.rs
parent989a4807b151f08c651b5027cc1b547a59adf966 (diff)
downloadcandle-c1453f00b11c9dd12c5aa81fb4355ce47d22d477.tar.gz
candle-c1453f00b11c9dd12c5aa81fb4355ce47d22d477.tar.bz2
candle-c1453f00b11c9dd12c5aa81fb4355ce47d22d477.zip
Improve the safetensor loading in the segment-anything example. (#772)
* Improve the safetensor loading in the segment-anything example. * Properly handle the labels when embedding the point prompts.
Diffstat (limited to 'candle-examples/examples/segment-anything/model_transformer.rs')
0 files changed, 0 insertions, 0 deletions