blip-caption icon indicating copy to clipboard operation
blip-caption copied to clipboard

Published 20 hours ago • verified publisher icon simonw

Added --gpu flag to run model on a gpu

Open mutherr opened this issue 4 months ago • 0 comments

I noticed there wasn't a way to run the model on a gpu, so added a flag to set the device for the pipeline.

mutherr avatar Sep 24 '24 18:09 mutherr