search

basetenlabs / truss-examples

Examples of models deployable with Truss
https://trussml.com
MIT License
143 stars 37 forks source link

added segment anything 2 #333

Open dsingal0 opened 3 months ago

dsingal0 commented 3 months ago

Restricted to segment-anything-2 small model for A10G. For larger checkpoints edit model.py and upgrade GPU. This model only generates the combined mask and individual masks for an input image.