imseldrith commited on
Commit
4c3d5e9
1 Parent(s): 8d2f66b

Update white_box_cartoonizer/cartoonize.py

Browse files
white_box_cartoonizer/cartoonize.py CHANGED
@@ -64,7 +64,7 @@ class WB_Cartoonize:
64
  gpu_options = None
65
  device_count = {'GPU':0}
66
 
67
- config = tf.ConfigProto(gpu_options=gpu_options, device_count=device_count)
68
 
69
  self.sess = tf.Session(config=config)
70
 
 
64
  gpu_options = None
65
  device_count = {'GPU':0}
66
 
67
+ config = tf.ConfigProto(gpu_options=None, device_count={'GPU':0})
68
 
69
  self.sess = tf.Session(config=config)
70