From 17fbe6b950b5ecf82584df6abc617290ccf10f0f Mon Sep 17 00:00:00 2001
From: Jason Antic <jsa169@gmail.com>
Date: Thu, 27 May 2021 16:48:07 -0700
Subject: [PATCH] Minor cpu performance fix and typo fix.

---
 deoldify/_device.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/deoldify/_device.py b/deoldify/_device.py
index 424e3d1..ed40ce1 100644
--- a/deoldify/_device.py
+++ b/deoldify/_device.py
@@ -13,7 +13,7 @@ class _Device:
 
     def is_gpu(self):
         ''' Returns `True` if the current device is GPU, `False` otherwise. '''
-        return self.current() is not DeviceID.CPU
+        return self.current() is not DeviceId.CPU
   
     def current(self):
         return self._current_device
@@ -26,6 +26,5 @@ class _Device:
             import torch
             torch.backends.cudnn.benchmark=False
         
-        os.environ['OMP_NUM_THREADS']='1'
         self._current_device = device    
         return device
\ No newline at end of file