File size: 274 Bytes
773f76d | 1 2 3 4 5 6 7 8 9 10 11 12 | from functools import lru_cache
import onnx
from facefusion.types import ModelInitializer
@lru_cache()
def get_static_model_initializer(model_path : str) -> ModelInitializer:
model = onnx.load(model_path)
return onnx.numpy_helper.to_array(model.graph.initializer[-1])
|