"transformer_engine/pytorch/quantization.py" did not exist on "2da34d4101905054890cb6bc0d3a2a7e468bb4ae"
Unverified Commit 90449f79 authored by Kirthi Shankar Sivamani's avatar Kirthi Shankar Sivamani Committed by GitHub
Browse files

Convert `NVFP4BlockScaling` to dataclass (#2227)



Fix passing args to nvfp4 recipe
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent f0a94048
......@@ -382,6 +382,7 @@ class Float8BlockScaling(Recipe):
)
@dataclass()
class NVFP4BlockScaling(Recipe):
"""
Use the NVFP4 scaling strategy.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment