Merge pull request #155 from ShivamKumar2002/patch-1

Fix GPU not being used even when it available
This commit is contained in:
Somdev Sangwan 2023-05-31 09:39:14 +05:30 committed by GitHub
commit 6416d4ebe8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,3 +1,4 @@
import torch
import onnxruntime
use_gpu = False