How to run this model with GPU?
#1
by
fudanjx
- opened
As title, How to run this model with GPU? I have successfully run it with CPU locally, works great, but think how can we speed up the processing with GPU?
I don't think that is model specific. The model can run on cpu and gpu, you need to specify which device to run the model on in your python scripts.
Could you give us more information on how you are using it? A code snippet maybe? Thanks!
prajwal967
changed discussion status to
closed