maduvantha commited on
Commit
e721b40
·
1 Parent(s): a2997fd

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -22,7 +22,7 @@ from tempfile import NamedTemporaryFile
22
  if sys.version_info[0] < 3:
23
  raise Exception("You must use Python 3 or higher. Recommended version is Python 3.7")
24
 
25
- def load_checkpoints(config_path, checkpoint_path, cpu=False):
26
 
27
  with open(config_path) as f:
28
  config = yaml.full_load(f)
 
22
  if sys.version_info[0] < 3:
23
  raise Exception("You must use Python 3 or higher. Recommended version is Python 3.7")
24
 
25
+ def load_checkpoints(config_path, checkpoint_path, cpu=True):
26
 
27
  with open(config_path) as f:
28
  config = yaml.full_load(f)