Spaces:
Runtime error
Runtime error
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:
|