Fabrice-TIERCELIN commited on
Commit
f650785
·
verified ·
1 Parent(s): 33c02fd

elif dit_weight.is_file() or dit_weight.is_symlink():

Browse files
Files changed (1) hide show
  1. hyvideo/inference.py +2 -1
hyvideo/inference.py CHANGED
@@ -317,6 +317,7 @@ class Inference(object):
317
  print('dit_weight.is_fifo(): ' + str(dit_weight.is_fifo()))
318
  print('dit_weight.is_block_device(): ' + str(dit_weight.is_block_device()))
319
  print('dit_weight.is_char_device(): ' + str(dit_weight.is_char_device()))
 
320
  if dit_weight.is_dir():
321
  files = list(dit_weight.glob("*.pt"))
322
  if len(files) == 0:
@@ -340,7 +341,7 @@ class Inference(object):
340
  f"`*_model_states.pt`(saved by deepspeed) can be parsed. If you want to load a "
341
  f"specific weight file, please provide the full path to the file."
342
  )
343
- elif dit_weight.is_file():
344
  model_path = dit_weight
345
  bare_model = "unknown"
346
  else:
 
317
  print('dit_weight.is_fifo(): ' + str(dit_weight.is_fifo()))
318
  print('dit_weight.is_block_device(): ' + str(dit_weight.is_block_device()))
319
  print('dit_weight.is_char_device(): ' + str(dit_weight.is_char_device()))
320
+ print('dit_weight.is_symlink(): ' + str(dit_weight.is_symlink()))
321
  if dit_weight.is_dir():
322
  files = list(dit_weight.glob("*.pt"))
323
  if len(files) == 0:
 
341
  f"`*_model_states.pt`(saved by deepspeed) can be parsed. If you want to load a "
342
  f"specific weight file, please provide the full path to the file."
343
  )
344
+ elif dit_weight.is_file() or dit_weight.is_symlink():
345
  model_path = dit_weight
346
  bare_model = "unknown"
347
  else: