Spaces:
Running
Running
Fabrice-TIERCELIN
commited on
elif dit_weight.is_file() or dit_weight.is_symlink():
Browse files- 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:
|