  
          self.device = self.set_device()
  
          if self.device == -1:
              # Limit ram usage to 3072 for CPU
              self.vram_total = 3072
          else:
              self.vram_total = self.stats.vram[self.device]
          
          self.get_available_vram()
  
 
</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 class GPUMem(object):
 </s> add class GPUMem():
            # Limit ram usage to 2048 for CPU
            self.vram_total = 2048
  
          self.device = self.set_device()
  
          if self.device == -1:
            # Limit ram usage to 2048 for CPU
            self.vram_total = 2048
            # Limit ram usage to 2048 for CPU
            self.vram_total = 2048
          else:
              self.vram_total = self.stats.vram[self.device]
          
          self.get_available_vram()
  
keep keep keep keep replace replace keep keep keep keep keep
 Extract: Limit non-gpu RAM useage to 2GB 
lib/face_alignment/vram_allocation.py