copy SAM model to args.device

yeungchenwa / OCR-SAM

Combining MMOCR with Segment Anything & Stable Diffusion. Automatically detect, recognize and segment text instances, with serval downstream tasks, e.g., Text Removal and Text Inpainting

533 stars 37 forks source link

copy SAM model to args.device #17

Closed flrngel closed 1 year ago

flrngel commented 1 year ago

This PR is to copy a SAM model to the user's desired device. Performs 6x faster on 3090.

Mountchicken commented 1 year ago

Hi @flrngel Thanks for your contribution!