File size: 288 Bytes
3914b35 |
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(maxsize = None)
def get_static_model_initializer(model_path : str) -> ModelInitializer:
model = onnx.load(model_path)
return onnx.numpy_helper.to_array(model.graph.initializer[-1])
|