MCPcopy
hub / github.com/NVIDIA/Stable-Diffusion-WebUI-TensorRT / get_onnx_path

Method get_onnx_path

model_manager.py:46–49  ·  view source on GitHub ↗
(model_name)

Source from the content-addressed store, hash-verified

44
45 @staticmethod
46 def get_onnx_path(model_name):
47 onnx_filename = f"{model_name}.onnx"
48 onnx_path = os.path.join(ONNX_MODEL_DIR, onnx_filename)
49 return onnx_filename, onnx_path
50
51 def get_trt_path(self, model_name, model_hash, profile, static_shape):
52 profile_hash = []

Callers 3

export_unet_to_trtFunction · 0.80
export_lora_to_trtFunction · 0.80
get_lorasMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected