  
  from lib.gpu_stats import GPUStats
  
  
  class GPUMem(object):
      """ Sets the scale to factor for dlib images
          and the ratio of vram to use for tensorflow """
  
      def __init__(self):
          self.initialized = False
 
</s> remove             # Limit ram usage to 3072 for CPU
            self.vram_free = 3072
        else:        
 </s> add             # Limit RAM to 2GB for non-gpu
            self.vram_free = 2048
        else: </s> remove             # Limit ram usage to 3072 for CPU
            self.vram_total = 3072
 </s> add             # Limit ram usage to 2048 for CPU
            self.vram_total = 2048
class GPUMem():
  
  from lib.gpu_stats import GPUStats
  
  
class GPUMem():
      """ Sets the scale to factor for dlib images
          and the ratio of vram to use for tensorflow """
  
      def __init__(self):
          self.initialized = False
keep keep keep keep replace keep keep keep keep keep
 Extract: Limit non-gpu RAM useage to 2GB 
lib/face_alignment/vram_allocation.py