Fabrice-TIERCELIN commited on
Commit
0871c8c
·
verified ·
1 Parent(s): 070c1ad

No more logs

Browse files
Files changed (1) hide show
  1. hyvideo/inference.py +1 -13
hyvideo/inference.py CHANGED
@@ -306,18 +306,6 @@ class Inference(object):
306
  f"specific weight file, please provide the full path to the file."
307
  )
308
  else:
309
- print('What is dit_weight: ' + str(dit_weight))
310
- print('dit_weight.exists(): ' + str(dit_weight.exists()))
311
- print('dit_weight.is_file(): ' + str(dit_weight.is_file()))
312
- print('dit_weight.is_dir(): ' + str(dit_weight.is_dir()))
313
- print('dit_weight.is_symlink(): ' + str(dit_weight.is_symlink()))
314
- print('dit_weight.is_junction(): ' + str(dit_weight.is_junction()))
315
- print('dit_weight.is_mount(): ' + str(dit_weight.is_mount()))
316
- print('dit_weight.is_socket(): ' + str(dit_weight.is_socket()))
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,7 +329,7 @@ class Inference(object):
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:
 
306
  f"specific weight file, please provide the full path to the file."
307
  )
308
  else:
 
 
 
 
 
 
 
 
 
 
 
 
309
  if dit_weight.is_dir():
310
  files = list(dit_weight.glob("*.pt"))
311
  if len(files) == 0:
 
329
  f"`*_model_states.pt`(saved by deepspeed) can be parsed. If you want to load a "
330
  f"specific weight file, please provide the full path to the file."
331
  )
332
+ elif dit_weight.is_file():
333
  model_path = dit_weight
334
  bare_model = "unknown"
335
  else: