Fabrice-TIERCELIN
commited on
No more logs
Browse files- 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()
|
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:
|