huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

TinyViT backbone for segment-anything. (#787)

Browse Source 
* TinyViT.

* More TinyViT.

* Add more to the tinyvit backbone.

* Proper padding.

* Plus ViT.

* Add the tiniest vit spec.
This commit is contained in:
Laurent Mazare
2023-09-09 15:10:06 +01:00
committed by GitHub
parent 3cd7e7b51d
commit b7cd58473b
3 changed files with 577 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-examples/examples/segment-anything/main.rs
View File

@ -11,6 +11,7 @@ pub mod model_image_encoder;
pub mod model_mask_decoder;
pub mod model_prompt_encoder;
pub mod model_sam;
pub mod model_tiny_vit;
pub mod model_transformer;
use candle::{DType, Result, Tensor};