cortexelus commited on
Commit
4ea87d6
Β·
verified Β·
1 Parent(s): 74cd2b7

Rename the autoencoder ONNX to match the engine naming scheme

Browse files

dec_*_chunkable_limiter.onnx -> dec_*_limiter.onnx
{enc,dec}_dynamic_triton_swa.onnx -> {enc,dec}_fp16.onnx (SAME-L)
{enc,dec}_dynamic_bf16.onnx -> {enc,dec}_bf16.onnx (SAME-S)

'chunkable' was never a property of the ONNX: it is the two optimization profiles
the TRT builder adds, so the engines keep the name and the graphs drop it.
'dynamic_triton_swa' named the samel::diff_attn_swa plugin node rather than a
precision, which put it out of step with every other file here.

Engine filenames are unchanged.

onnx/same-l/{dec_dynamic_triton_swa.onnx β†’ dec_fp16.onnx} RENAMED
File without changes
onnx/same-l/{dec_fp16_chunkable_limiter.onnx β†’ dec_fp16_limiter.onnx} RENAMED
File without changes
onnx/same-l/{dec_fp8_chunkable_limiter.onnx β†’ dec_fp8_limiter.onnx} RENAMED
File without changes
onnx/same-l/{enc_dynamic_triton_swa.onnx β†’ enc_fp16.onnx} RENAMED
File without changes
onnx/same-s/{dec_dynamic_bf16.onnx β†’ dec_bf16.onnx} RENAMED
File without changes
onnx/same-s/{dec_bf16_chunkable_limiter.onnx β†’ dec_bf16_limiter.onnx} RENAMED
File without changes
onnx/same-s/{dec_fp8_chunkable_limiter.onnx β†’ dec_fp8_limiter.onnx} RENAMED
File without changes
onnx/same-s/{enc_dynamic_bf16.onnx β†’ enc_bf16.onnx} RENAMED
File without changes