sahi icon indicating copy to clipboard operation
sahi copied to clipboard

fix: specify a device other than cuda:0

Open 0xf21 opened this issue 7 months ago • 1 comments

I modified a few lines of code to specify other devices, such as cuda:1 or cpu, instead of only selecting cuda:0 when instantiating a detection model using the AutoDetectionModel.from_pretrained method.

0xf21 avatar May 21 '25 07:05 0xf21

@0xf21 perfect addition, thanks!

fcakyon avatar May 24 '25 21:05 fcakyon