[Bugfix] Fix GPT-OSS on AMD after #28603 (#28816)

Signed-off-by: zhewenli <zhewenli@meta.com>
This commit is contained in:
Zhewen Li
2025-11-17 10:15:26 -08:00
committed by GitHub
parent e42bd8c2e3
commit f8b19c0ffd
2 changed files with 7 additions and 6 deletions

View File

@@ -755,8 +755,8 @@ class Mxfp4MoEMethod(FusedMoEMethodBase):
self.w13_weight = w13_weight
self.w2_weight = w2_weight
layer.w13_weight = Parameter(w13_weight.data, requires_grad=False)
layer.w2_weight = Parameter(w2_weight.data, requires_grad=False)
layer.w13_weight = Parameter(w13_weight.storage.data, requires_grad=False)
layer.w2_weight = Parameter(w2_weight.storage.data, requires_grad=False)
else:
raise ValueError(f"Unsupported backend: {self.mxfp4_backend}")