13 lines
287 B
Python
13 lines
287 B
Python
import os
|
|
|
|
import torch
|
|
|
|
from fastvideo.models.stepvideo.config import parse_args
|
|
|
|
try:
|
|
args = parse_args()
|
|
torch.ops.load_library(
|
|
os.path.join(args.model_dir, 'lib/liboptimus_ths-torch2.5-cu124.cpython-310-x86_64-linux-gnu.so'))
|
|
except Exception as err:
|
|
print(err)
|