TinyViT backbone for segment-anything. (#787)

* TinyViT. * More TinyViT. * Add more to the tinyvit backbone. * Proper padding. * Plus ViT. * Add the tiniest vit spec.
huggingface · Sep 9, 2023 · b7cd584 · b7cd584
1 parent 3cd7e7b
commit b7cd584
Show file tree

Hide file tree

Showing 3 changed files with 577 additions and 13 deletions.
diff --git a/candle-examples/examples/segment-anything/main.rs b/candle-examples/examples/segment-anything/main.rs
@@ -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};