mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-23 01:46:09 +02:00
henryruhs
12 lines
288 B
Python
12 lines
288 B
Python
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])
|