(quant_detected, quant_op_val)
| 463 | |
| 464 | |
| 465 | def get_quant_scheme(quant_detected, quant_op_val): |
| 466 | if quant_op_val == "torch": |
| 467 | return f"{quant_detected}-torchao" |
| 468 | elif quant_op_val == "triton": |
| 469 | return f"{quant_detected}-triton" |
| 470 | else: |
| 471 | return f"{quant_detected}-{quant_op_val}" |
| 472 | |
| 473 | |
| 474 | def build_wan21( |
no outgoing calls
no test coverage detected