Spaces:
Running
Running
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:
|